def rename_group(self, new_group_name):
db = DBUtil() # 创建数据库对象
db.connect_db() # 创建连接对象
db.cur_db() # 创建游标
sql_stm = "update friend_group set friend_group_name='{}' where friend_group_user_id={}"\
.format(new_group_name, self.group_user_id)
result = db.exc_sql(sql_stm)
return result
def del_friend_group(self):
db = DBUtil() # 创建数据库对象
db.connect_db() # 创建连接对象
db.cur_db() # 创建游标
sql_stm = "delete from friend_group where friend_group_name='{}' and friend_group_user_id={}"\
.format(self.group_name, self.group_user_id)
result = db.exc_sql(sql_stm)
return result
def add_friend_group(self):
db = DBUtil() # 创建数据库对象
db.connect_db() # 创建连接对象
db.cur_db() # 创建游标
sql_stm = "insert into friend_group(friend_group_name,friend_group_user_id) " \
"values('{}',{})".format(self.group_name, self.group_user_id)
result = db.exc_sql(sql_stm)
return result
def del_friend(self):
if self.friend_id is None or self.friend_id == self.user_id:
return False
db = DBUtil() # 创建数据库对象
db.connect_db() # 创建连接对象
db.cur_db() # 创建游标
sql_stm = "delete from friendship_info where (friendship_user_id={} and friendship_friend_id={}) or "\
"(friendship_user_id={} and friendship_friend_id={})"\
.format(self.user_id, self.friend_id, self.friend_id, self.user_id)
result = db.exc_sql(sql_stm)
return result
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_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_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
