Compare commits

...

5 Commits

Author SHA1 Message Date
c9b6ce8f60 fix: Typo in community schema. 2025-09-04 01:08:32 +10:00
0363dc2bcd Catch IndexError for user upgrade. 2025-09-02 09:04:14 +10:00
ce46bd49fe fix: Remove transaction block. 2025-09-02 08:57:04 +10:00
4653689e5e fix: Profile component typo. 2025-09-02 07:55:41 +10:00
e8e44b7da8 Add twitch_setup export. 2025-09-02 07:54:08 +10:00
4 changed files with 27 additions and 29 deletions

View File

@@ -1 +1 @@
from .profiles import setup from .profiles import setup, twitch_setup

View File

@@ -41,7 +41,7 @@ CREATE UNIQUE INDEX profiles_twitch_userid ON profiles_twitch (userid);
CREATE TABLE communities( CREATE TABLE communities(
communityid INTEGER PRIMARY KEY GENERATED ALWAYS AS IDENTITY, communityid INTEGER PRIMARY KEY GENERATED ALWAYS AS IDENTITY,
migrated INTEGER REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE, migrated INTEGER REFERENCES communities (communityid) ON DELETE CASCADE ON UPDATE CASCADE,
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(), created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
last_seen TIMESTAMPTZ NOT NULL DEFAULT NOW(), last_seen TIMESTAMPTZ NOT NULL DEFAULT NOW(),
_timestamp TIMESTAMPTZ NOT NULL DEFAULT NOW() _timestamp TIMESTAMPTZ NOT NULL DEFAULT NOW()

View File

@@ -6,5 +6,5 @@ if TYPE_CHECKING:
async def setup(bot: 'Bot'): async def setup(bot: 'Bot'):
from .component import ProfileComponent from .component import ProfilesComponent
await bot.add_component(ProfileComponent(bot)) await bot.add_component(ProfilesComponent(bot))

View File

@@ -41,9 +41,7 @@ class ProfilesComponent(cmds.Component):
Fetch or create the profile for the given user. Fetch or create the profile for the given user.
""" """
userid = str(user.id) userid = str(user.id)
async with self.bot.dbconn.connection() as conn: # TODO: Transaction
self.bot.dbconn.conn = conn
async with conn.transaction():
profile = await self.profiles.get_profile_twitch(userid) profile = await self.profiles.get_profile_twitch(userid)
if profile is None: if profile is None:
args = {} args = {}
@@ -53,6 +51,8 @@ class ProfilesComponent(cmds.Component):
args['avatar'] = user.profile_image.url args['avatar'] = user.profile_image.url
except twitchio.HTTPException: except twitchio.HTTPException:
pass pass
except IndexError:
pass
profile = await UserProfile.create(**args) profile = await UserProfile.create(**args)
await TwitchProfileLink.create(profileid=profile.profileid, userid=userid) await TwitchProfileLink.create(profileid=profile.profileid, userid=userid)
elif touch: elif touch:
@@ -70,9 +70,7 @@ class ProfilesComponent(cmds.Component):
Fetch or create the community for this channel. Fetch or create the community for this channel.
""" """
chanid = channel.id chanid = channel.id
async with self.bot.dbconn.connection() as conn: # TODO: Transaction
self.bot.dbconn.conn = conn
async with conn.transaction():
comm = await self.profiles.get_community_twitch(chanid) comm = await self.profiles.get_community_twitch(chanid)
if comm is None: if comm is None:
comm = await Community.create() comm = await Community.create()