107 lines
2.5 KiB
Python
107 lines
2.5 KiB
Python
from data import Registry, RowModel, Table
|
|
from data.columns import String, Timestamp, Integer, Bool
|
|
|
|
|
|
class UserAuth(RowModel):
|
|
"""
|
|
Schema
|
|
======
|
|
CREATE TABLE user_auth(
|
|
userid TEXT PRIMARY KEY,
|
|
token TEXT NOT NULL,
|
|
refresh_token TEXT NOT NULL,
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
_timestamp TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
|
);
|
|
"""
|
|
_tablename_ = 'user_auth'
|
|
_cache_ = {}
|
|
|
|
userid = String(primary=True)
|
|
token = String()
|
|
refresh_token = String()
|
|
created_at = Timestamp()
|
|
_timestamp = Timestamp()
|
|
|
|
|
|
class BotChannel(RowModel):
|
|
"""
|
|
Schema
|
|
======
|
|
CREATE TABLE bot_channels(
|
|
userid TEXT PRIMARY KEY REFERENCES user_auth(userid) ON DELETE CASCADE,
|
|
autojoin BOOLEAN DEFAULT true,
|
|
listen_redeems BOOLEAN,
|
|
joined_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
_timestamp TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
|
);
|
|
"""
|
|
_tablename_ = 'bot_channels'
|
|
_cache_ = {}
|
|
|
|
userid = String(primary=True)
|
|
autojoin = Bool()
|
|
listen_redeems = Bool()
|
|
joined_at = Timestamp()
|
|
_timestamp = Timestamp()
|
|
|
|
|
|
class UserProfile(RowModel):
|
|
"""
|
|
Schema
|
|
======
|
|
CREATE TABLE user_profiles(
|
|
profileid INTEGER GENERATED ALWAYS AS IDENTITY PRIMARY KEY,
|
|
twitchid TEXT NOT NULL,
|
|
name TEXT,
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
_timestamp TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
|
);
|
|
"""
|
|
_tablename_ = 'user_profiles'
|
|
_cache_ = {}
|
|
|
|
profileid = Integer(primary=True)
|
|
twitchid = String()
|
|
name = String()
|
|
created_at = Timestamp()
|
|
_timestamp = Timestamp()
|
|
|
|
|
|
class Communities(RowModel):
|
|
"""
|
|
Schema
|
|
======
|
|
CREATE TABLE communities(
|
|
communityid INTEGER GENERATED ALWAYS AS IDENTITY PRIMARY KEY,
|
|
twitchid TEXT NOT NULL,
|
|
name TEXT,
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
_timestamp TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
|
);
|
|
"""
|
|
_tablename_ = 'communities'
|
|
_cache_ = {}
|
|
|
|
communityid = Integer(primary=True)
|
|
twitchid = Integer()
|
|
name = String()
|
|
created_at = Timestamp()
|
|
_timestamp = Timestamp()
|
|
|
|
|
|
class BotData(Registry):
|
|
user_auth = UserAuth.table
|
|
|
|
"""
|
|
CREATE TABLE user_auth_scopes(
|
|
userid TEXT NOT NULL REFERENCES user_auth(userid) ON DELETE CASCADE,
|
|
scope TEXT NOT NULL
|
|
);
|
|
"""
|
|
user_auth_scopes = Table('user_auth_scopes')
|
|
|
|
bot_channels = BotChannel.table
|
|
user_profiles = UserProfile.table
|
|
communities = Communities.table
|