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] # 用户注册时间，非空
