def get_user_info(cls, in_user_id): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 user_info_dict = dict() sql_stm = "select user_name,user_nick_name,user_signature,user_gender_id,user_birthday," \ "user_telephone,user_email,user_introduce,user_head,user_age,user_vocation," \ "user_nation_id,user_province_id,user_city_id,user_register_time " \ "from user_info where user_id={}".format(in_user_id) result = db.get_res(sql_stm) if result: user_info_dict['name'] = result[0][0] # 用户名 user_info_dict['nick_name'] = result[0][1] # 用户昵称 user_info_dict['signature'] = result[0][2] # 用户签名 user_info_dict['gender_id'] = result[0][3] # 用户性别,1为男,2为女,3为保密 if result[0][4] is None: user_info_dict['birthday'] = '' user_info_dict['age'] = 0 else: user_info_dict['birthday'] = str(result[0][4]) # 用户生日 user_info_dict['age'] = localtime(time())[0] - int( str(result[0][4]).split('-')[0]) # 用户年龄 user_info_dict['telephone'] = result[0][5] # 设置用户登录手机号,非空 user_info_dict['email'] = result[0][6] # 设置登录邮箱,非空 # self.introduce = result[0][7] # 用户介绍bbb # self.head = result[0][8] # 用户头像 user_info_dict['vocation'] = result[0][10] # 用户的职业信息 # self.nation_id = result[0][11] # 用户所属国家 # self.province_id = result[0][12] # 用户所属省 # self.city_id = result[0][13] # 用户所属城市 user_info_dict['register_time'] = str( result[0][14].date()) # 用户注册时间,非空 return str(user_info_dict)
def get_remark_name(self): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select friendship_name from friendship_info where friendship_user_id={} and " \ "friendship_friend_id={}".format(self.user_id, self.friend_id) result = db.get_res(sql_stm) self.remark_name = result[0][0] return self.remark_name
def get_status(self): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select user_status_id from user_info " \ "where user_name='{}'".format(self.name) result = db.get_res(sql_stm) if result: self.status_id = result[0][0] return self.status_id
def get_group_info(cls, group_name): db = DBUtil() db.connect_db() db.cur_db() sql_stm = "select group_id,group_admin_id,group_name from group_info where group_name='{}'".format( group_name) result = db.get_res(sql_stm) if result: return result[0].__str__() else: return ""
def get_all_group(cls, in_user_id): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select friend_group_name from friend_group where friend_group_user_id={}".format( in_user_id) result = db.get_res(sql_stm) group_list = [] for group in result: group_list.append(group[0]) return group_list
def get_init_group_id(friend_group_user_id): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select friend_group_id from friend_group where friend_group_name='{}' " \ "and friend_group_user_id={}".format("我的好友", friend_group_user_id) result = db.get_res(sql_stm) if result: return result[0][0] else: return "null"
def get_friend_id_list(cls, in_user_id): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select friendship_friend_id from friendship_info where friendship_user_id={}".format( in_user_id) result = db.get_res(sql_stm) friend_id_list = [] for friend_id in result: friend_id_list.append(friend_id[0]) return str(friend_id_list)
def find_id(self): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select user_id from user_info where user_name='{}'".format( self.friend_name) result = db.get_res(sql_stm) if result: return result[0][0] else: return None
def get_friend_group_name(cls, in_group_id): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select friend_group_name from friend_group where friend_group_id={}".format( in_group_id) result = db.get_res(sql_stm) if result: return result[0][0] else: return None
def get_friend_list(cls, in_user_id): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select friendship_friend_id,friendship_name from friendship_info " \ "where friendship_user_id={}".format(in_user_id) result = db.get_res(sql_stm) friend_list = [] for friend in result: friend_mark_name = User.id_to_name(friend[0]) friend_list.append((friend[0], friend_mark_name, friend[1])) return str(friend_list)
def get_friend_group_id(self): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select friend_group_id from friend_group where friend_group_user_id={}".format( self.group_user_id) result = db.get_res(sql_stm) if result: self.group_user_id = result[0][0] else: self.group_user_id = None return self.group_user_id
def id_to_name(cls, in_user_id): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select user_name from user_info where user_id={}".format( in_user_id) result = db.get_res(sql_stm) if result: out_user_name = result[0][0] else: out_user_name = None return out_user_name
def get_group(self): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select friendship_friend_group_id from friendship_info where friendship_user_id={} and " \ "friendship_friend_id={}".format(self.user_id, self.friend_id) result = db.get_res(sql_stm) if result: self.friend_group = result[0][0] else: self.friend_group = "null" return self.friend_group
def get_user_id(self): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 args = (self.name, ) sql_stm = "select user_id from user_info where user_name=%s" result = db.get_res(sql_stm, *args) if result: self.id = result[0][0] else: self.id = None return self.id
def name_to_id(cls, in_user_name): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 args = (in_user_name, in_user_name, in_user_name) # 设置SQL语句参数 sql_stm = "select user_id from user_info where " \ "(user_name=%s or user_email=%s or user_telephone=%s)" result = db.get_res(sql_stm, *args) if result: out_user_id = result[0][0] else: out_user_id = None return out_user_id
def get_all_friend(cls, in_user_id): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select friendship_friend_id,friendship_friend_group_id from friendship_info " \ "where friendship_user_id={}".format(in_user_id) result = db.get_res(sql_stm) friend_dict = {} for friend in result: friend_group_name = FriendGroup.get_friend_group_name(friend[1]) friend_name = User.id_to_name(friend[0]) if friend_group_name not in friend_dict: friend_dict[friend_group_name] = [] friend_dict[friend_group_name].append(friend_name) return friend_dict.__str__()
def login(self): __doc__ = ''' 用户登录 :return: 登录是否成功 ''' db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 args = (self.password, self.name, self.name, self.name) # 设置SQL语句参数 sql_stm = "select user_id from user_info where (user_password=%s and " \ "(user_name=%s or user_email=%s or user_telephone=%s))" result = db.get_res(sql_stm, *args) # 执行SQL语句 if result: self.id = result[0][0] self.get_inf() return self.id # 获取用户ID else: return None
def get_inf(self): db = DBUtil() # 创建数据库对象 db.connect_db() # 创建连接对象 db.cur_db() # 创建游标 sql_stm = "select user_name,user_nick_name,user_signature,user_gender_id,user_birthday," \ "user_telephone,user_email,user_introduce,user_head,user_age,user_vocation," \ "user_nation_id,user_province_id,user_city_id,user_register_time " \ "from user_info where user_id={}".format(self.id) result = db.get_res(sql_stm) self.name = result[0][0] self.nick_name = result[0][1] # 用户昵称 self.signature = result[0][2] # 用户签名 self.gender_id = result[0][3] # 用户性别,0为男,1为女 self.birthday = result[0][4] # 用户生日 self.telephone = result[0][5] # 设置用户登录手机号,非空 self.email = result[0][6] # 设置登录邮箱,非空 self.introduce = result[0][7] # 用户介绍 self.head = result[0][8] # 用户头像 self.age = result[0][9] # 用户年龄 self.vocation = result[0][10] # 用户的职业信息 self.nation_id = result[0][11] # 用户所属国家 self.province_id = result[0][12] # 用户所属省 self.city_id = result[0][13] # 用户所属城市 self.register_time = result[0][14] # 用户注册时间,非空