def add_friend(token, ftk):
friend = session.query(Friend).filter(Friend.ftk == ftk,
Friend.token == token).first()
if friend is not None:
return -1, friend
my_category = session.query(User).filter(
User.token == token).first().category
if my_category == 2:
count = session.query(func.count(
Friend.id)).filter(Friend.token == token).scalar()
if count > max_helpers:
return -2, None
user = session.query(User).filter(User.token == ftk).first()
if user is None:
return -3, None
friend_category = user.category
if friend_category == 2:
count = session.query(func.count(Friend.id)).filter(
Friend.ftk == ftk, Friend.category == 2).scalar()
if count > max_helpers:
return -2, None
if my_category == 2 and friend_category == 2:
return -4, None
intimacy = 0
if my_category == 1 and friend_category == 2:
intimacy = 1
adt = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
friend = Friend(token, ftk, friend_category, intimacy, adt)
session.add(friend)
friend = Friend(ftk, token, my_category, 0, adt)
session.add(friend)
session.commit()
friend = session.query(Friend).filter(Friend.ftk == ftk,
Friend.token == token).first()
return 0, friend
def update_sn(token, sn):
try:
session.query(User).filter(User.token == token).update({User.sn: sn})
session.commit()
except InvalidRequestError:
session.rollback()
except Exception as e:
print(e)
def update_intimacy(token, ftk, intimacy):
try:
session.query(Friend).filter(Friend.ftk == ftk,
Friend.token == token).update(
{Friend.intimacy: intimacy})
session.commit()
except InvalidRequestError:
session.rollback()
def do_save(code, content, url, image):
try:
ut = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
qrcode = Qrcode(code, content, url, image, ut)
session.add(qrcode)
session.commit()
except InvalidRequestError:
session.rollback()
def add_new_version_file(file_name, version, check_sum):
adt = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
try:
firmware = Firmware(file_name, version, check_sum, adt)
session.add(firmware)
session.commit()
except InvalidRequestError:
session.rollback()
async def update_token(old_token, new_token):
try:
session.query(Friend).filter(Friend.ftk == old_token).update(
{Friend.ftk: new_token})
session.query(Friend).filter(Friend.token == old_token).update(
{Friend.token: new_token})
session.commit()
except InvalidRequestError:
session.rollback()
def del_friend(token, ftk):
try:
session.query(Friend).filter(Friend.ftk == ftk,
Friend.token == token).delete()
session.query(Friend).filter(Friend.token == ftk,
Friend.ftk == token).delete()
session.commit()
except InvalidRequestError:
session.rollback()
pass
def get_reg_user(number, password):
user = session.query(User).filter(User.number == number, User.password == password).first()
otk = ""
if user is not None:
otk = user.token
user.token = Encode.generate_token()
user.lt = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
session.query(User).filter(User.number == number).update({User.token: user.token, User.lt: user.lt})
session.commit()
return otk, user
def add_user(token, category):
user = session.query(User).filter(User.token == token).first()
if user is not None:
return -1, user
rt = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
user = User("", "", "", token, "", category, "", rt, rt)
session.add(user)
session.commit()
user = session.query(User).filter(User.token == token).first()
return 0, user
def update_user(token, name, image):
user = None
try:
session.query(User).filter(User.token == token).update({User.name: name, User.image: image})
session.commit()
user = session.query(User).filter(User.token == token).first()
except InvalidRequestError:
session.rollback()
except Exception as e:
print(e)
return user
def reg_user(number, password, category, dev_sn):
user = session.query(User).filter(User.number == number).first()
if user is not None:
return -1, user
rt = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
token = Encode.generate_token()
user = User("", number, password, token, "", category, dev_sn, rt, rt)
session.add(user)
session.commit()
user = session.query(User).filter(User.token == token).first()
return 0, user
def add_new_version(project, version_name, base_code, version_code,
file_type, file_name, apk_check_sum, patch_check_sum):
adt = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
try:
version = Version(project, version_name, base_code, version_code,
file_type, file_name, apk_check_sum,
patch_check_sum, adt)
session.add(version)
session.commit()
except InvalidRequestError:
session.rollback()
