新闻推荐系统第一版代码

codes/news_rec_server/controller/log_controller.py

@@ -0,0 +1,19 @@
+from dao.mysql_server import MysqlServer
+from dao.entity.logitem import LogItem
+
+import time
+
+
+
+class LogController():
+    def __init__(self) -> None:
+        self.log_info_sql_session = MysqlServer().get_loginfo_session()
+
+    def save_one_log(self,log):
+        try:
+            self.log_info_sql_session.add(log)
+            self.log_info_sql_session.commit()
+        except Exception as e:
+            print(str(e))
+            return False
+        return True

codes/news_rec_server/controller/user_action_controller.py

@@ -0,0 +1,116 @@
+
+from dao.mysql_server import MysqlServer
+from dao.entity.user_exposure import UserExposure
+from dao.entity.register_user import RegisterUser
+from dao.entity.user_likes import UserLikes
+from dao.entity.user_read import UserRead
+from dao.entity.user_collections import UserCollections
+
+# 初始化数据表
+user = UserLikes()
+user = UserCollections()
+user = UserExposure()
+user = UserRead()
+
+
+class UserAction():
+    def __init__(self) -> None:
+        self.user_exposure_sql_session = MysqlServer().get_user_exposure_session()
+        self.register_user_sql_session = MysqlServer().get_register_user_session() 
+        self.user_like_sql_session = MysqlServer().get_user_like_session()
+        self.user_collection_sql_session = MysqlServer().get_user_collection_session()
+        self.user_read_sql_session = MysqlServer().get_user_read_session()
+
+    def user_is_exist(self, user, user_type):
+        """
+        1 表示正确；2 表示密码错误；0 表示用户不存在
+        """
+        if user_type == "login":
+            if self.register_user_sql_session.query(RegisterUser).filter(RegisterUser.username == user.username, \
+                RegisterUser.passwd == user.passwd).count() > 0:
+                return 1
+            elif self.register_user_sql_session.query(RegisterUser).filter(RegisterUser.username == user.username).count() > 0:
+                return 2
+            else:
+                return 0
+        else:
+            if self.register_user_sql_session.query(RegisterUser).filter(RegisterUser.username == user.username).count() > 0:
+                return 1
+            else:
+                return 0
+
+    def save_user(self,user):
+        try:
+            self.register_user_sql_session.add(user)
+            self.register_user_sql_session.commit()
+            # self.register_user_sql_session.close()
+        except Exception as e:
+            print(str(e))
+            return False
+        return True
+    
+    def get_user_id_by_name(self,username):
+
+        try:
+            userid = self.register_user_sql_session.query(RegisterUser.userid).filter(RegisterUser.username == username).one()[0]
+        except Exception as e:
+            print(str(e))
+            return None
+        return userid
+
+    def get_likes_counts_by_user(self,user_id,news_id):
+        return self.user_like_sql_session.query(UserLikes).filter(UserLikes.userid == user_id, UserLikes.newid == news_id).count()
+
+    def get_coll_counts_by_user(self,user_id,news_id):
+        return self.user_collection_sql_session.query(UserCollections).filter(UserCollections.userid == user_id,
+                UserCollections.newid == news_id).count()
+    
+    def del_likes_by_user(self,user_id,news_id):
+        try:
+            print(user_id,news_id)
+            delItems = self.user_like_sql_session.query(UserLikes).filter(UserLikes.userid == user_id, UserLikes.newid == news_id)
+            # print(delItems.count())
+            if delItems.count() > 0:
+                self.user_like_sql_session.query(UserLikes).filter(UserLikes.userid == user_id, UserLikes.newid == news_id).delete()
+                self.user_like_sql_session.commit()
+        except Exception as e:
+            print(str(e))
+            return False
+        return True
+    
+    def del_coll_by_user(self,user_id,news_id):
+        try:
+            delItems = self.user_collection_sql_session.query(UserCollections).filter(UserCollections.userid == user_id, 
+                    UserCollections.newid == news_id)
+            if delItems.count() > 0:
+                self.user_collection_sql_session.query(UserCollections).filter(UserCollections.userid == user_id, 
+                    UserCollections.newid == news_id).delete()
+                self.user_collection_sql_session.commit()
+        except Exception as e:
+            print(str(e))
+            return False
+        return True
+
+    def save_one_action(self,action):
+        if isinstance(action, UserLikes):
+            try:
+                self.user_like_sql_session.add(action)
+                self.user_like_sql_session.commit()
+            except Exception as e:
+                print(str(e))
+                return False
+        elif isinstance(action, UserCollections):
+            try:
+                self.user_collection_sql_session.add(action)
+                self.user_collection_sql_session.commit()
+            except Exception as e:
+                print(str(e))
+                return False
+        elif isinstance(action, UserRead):
+            try:
+                self.user_read_sql_session.add(action)
+                self.user_read_sql_session.commit()
+            except Exception as e:
+                print(str(e))
+                return False  
+        return True