1
0

user.py 1.1 KB

12345678910111213141516171819202122232425262728
  1. async def create_user(pg, user_id):
  2. await pg.execute("INSERT INTO \"user\"(user_id) VALUES($1) ON CONFLICT DO NOTHING", user_id)
  3. async def ignore_user(pg, user_id):
  4. await pg.execute("UPDATE \"user\" SET ignore = TRUE WHERE user_id = $1", user_id)
  5. async def unignore_user(pg, user_id):
  6. await pg.execute("UPDATE \"user\" SET ignore = FALSE WHERE user_id = $1", user_id)
  7. async def is_ignored(pg, user_id):
  8. return await pg.fetchval("SELECT ignore FROM \"user\" WHERE user_id = $1", user_id)
  9. async def get_level(pg, user_id):
  10. return await pg.fetchval("SELECT level FROM \"user\" WHERE user_id = $1", user_id)
  11. async def get_xp(pg, user_id):
  12. message_array = await pg.fetch("SELECT total_messages FROM channel_user WHERE user_id = $1", user_id)
  13. print(message_array)
  14. total_xp = 0
  15. for messages in message_array:
  16. print(messages)
  17. total_xp += messages
  18. xp_spent = await pg.fetchval("SELECT xp_spent FROM \"user\" WHERE user_id = $1", user_id)
  19. return xp_spent, total_xp
  20. async def level_up(pg, user_id, xp):
  21. await pg.execute("UPDATE \"user\" (level, xp_spent) VALUES(+ 1, $2) WHERE user_id = $1", user_id, xp)