Conatum
ac71c4da9b
New `study_time_since` database function from session history. Add `Lion.timezone`. Add `Lion.day_start`. Add `Lion.studied_today`. Made `Table.queries` an instance variable. Renamed the session channel types.
90 lines
2.1 KiB
Python
90 lines
2.1 KiB
Python
from psycopg2.extras import execute_values
|
|
|
|
from cachetools import TTLCache
|
|
from data import RowTable, Table
|
|
|
|
|
|
meta = RowTable(
|
|
'AppData',
|
|
('appid', 'last_study_badge_scan'),
|
|
'appid',
|
|
attach_as='meta',
|
|
)
|
|
|
|
|
|
user_config = RowTable(
|
|
'user_config',
|
|
('userid', 'timezone'),
|
|
'userid',
|
|
cache=TTLCache(5000, ttl=60*5)
|
|
)
|
|
|
|
|
|
guild_config = RowTable(
|
|
'guild_config',
|
|
('guildid', 'admin_role', 'mod_role', 'event_log_channel', 'alert_channel',
|
|
'min_workout_length', 'workout_reward',
|
|
'max_tasks', 'task_reward', 'task_reward_limit',
|
|
'study_hourly_reward', 'study_hourly_live_bonus',
|
|
'study_ban_role', 'max_study_bans'),
|
|
'guildid',
|
|
cache=TTLCache(1000, ttl=60*5)
|
|
)
|
|
|
|
unranked_roles = Table('unranked_roles')
|
|
|
|
donator_roles = Table('donator_roles')
|
|
|
|
|
|
lions = RowTable(
|
|
'members',
|
|
('guildid', 'userid',
|
|
'tracked_time', 'coins',
|
|
'workout_count', 'last_workout_start',
|
|
'last_study_badgeid',
|
|
'video_warned',
|
|
'_timestamp'
|
|
),
|
|
('guildid', 'userid'),
|
|
cache=TTLCache(5000, ttl=60*5),
|
|
attach_as='lions'
|
|
)
|
|
|
|
lion_ranks = Table('member_ranks', attach_as='lion_ranks')
|
|
|
|
|
|
@lions.save_query
|
|
def add_pending(pending):
|
|
"""
|
|
pending:
|
|
List of tuples of the form `(userid, pending_coins, pending_time)`.
|
|
"""
|
|
with lions.conn:
|
|
cursor = lions.conn.cursor()
|
|
data = execute_values(
|
|
cursor,
|
|
"""
|
|
UPDATE members
|
|
SET
|
|
coins = coins + t.coin_diff,
|
|
tracked_time = tracked_time + t.time_diff
|
|
FROM
|
|
(VALUES %s)
|
|
AS
|
|
t (guildid, userid, coin_diff, time_diff)
|
|
WHERE
|
|
members.guildid = t.guildid
|
|
AND
|
|
members.userid = t.userid
|
|
RETURNING *
|
|
""",
|
|
pending,
|
|
fetch=True
|
|
)
|
|
return lions._make_rows(*data)
|
|
|
|
|
|
global_guild_blacklist = Table('global_guild_blacklist')
|
|
global_user_blacklist = Table('global_user_blacklist')
|
|
ignored_members = Table('ignored_members')
|