def send(uid, tpuid):
db = Connection()
sql_0 = f"""SELECT * FROM `friends` WHERE (`uid`={uid} AND `tpuid`={tpuid})
OR (`uid`={tpuid} AND `tpuid`={uid}) AND flag=0"""
sql_1 = f"""SELECT * FROM `friends` WHERE (`uid`={uid} AND `tpuid`={tpuid})
OR (`uid`={tpuid} AND `tpuid`={uid}) AND flag=1"""
if (db.check(sql_0)):
return -1
elif (db.check(sql_1)):
return -3
sql = """
INSERT INTO `friends` (`uid`, `tpuid`)
VALUES
( %s, %s )
""" % (uid, tpuid)
return db.set(sql)
def VsOb(uid, tpuid):
# friendship status check
db = Connection()
sql_0 = f"""SELECT * FROM `friends` WHERE ((`uid`={uid} AND `tpuid`={tpuid})
OR (`uid`={tpuid} AND `tpuid`={uid})) AND flag=0"""
sql_1 = f"""SELECT * FROM `friends` WHERE ((`uid`={uid} AND `tpuid`={tpuid})
OR (`uid`={tpuid} AND `tpuid`={uid})) AND flag=1"""
sql_2 = f"""SELECT * FROM `friends` WHERE (`uid`={uid} AND `tpuid`={tpuid}) AND flag=0"""
sql_3 = f"""SELECT * FROM `friends` WHERE (`uid`={tpuid} AND `tpuid`={uid}) AND flag=0"""
if (db.check(sql_0)):
if (db.check(sql_2)):
return -10
else:
return -11
return -1
elif (db.check(sql_1)):
return -3
return 0
def login(self, email, pwd):
_pwd = hashlib.sha256(pwd.encode()).hexdigest()
sql = """
SELECT * FROM users WHERE email='%s' AND pwd='%s'
""" % (email, _pwd)
obj = Connection()
if (obj.check(sql)):
return obj.get(sql)[0]
else:
return False
def name(uid):
db = Connection()
sql = f"SELECT * FROM users WHERE id={uid}"
return db.get(sql)[0][1] if db.check(sql) == True else "Anonymous"
