From a9a8339d2abd485d587080e31c3e35e1a26840db Mon Sep 17 00:00:00 2001 From: tdwaf Date: Sat, 30 Nov 2024 19:56:06 -0500 Subject: [PATCH] feat: ttttt --- kusogaki_bot/data/food_counter_repository.py | 5 +++-- kusogaki_bot/data/models.py | 3 ++- kusogaki_bot/data/reminder_repository.py | 5 +++-- kusogaki_bot/data/scheduled_thread_repository.py | 5 +++-- 4 files changed, 11 insertions(+), 7 deletions(-) diff --git a/kusogaki_bot/data/food_counter_repository.py b/kusogaki_bot/data/food_counter_repository.py index 4b90e58..f031716 100644 --- a/kusogaki_bot/data/food_counter_repository.py +++ b/kusogaki_bot/data/food_counter_repository.py @@ -1,10 +1,11 @@ import logging from datetime import datetime -from db import Database -from models import FoodCounter from sqlalchemy.exc import SQLAlchemyError +from kusogaki_bot.data.db import Database +from kusogaki_bot.data.models import FoodCounter + class FoodCounterRepository: """Repository class for food counter persistence using PostgreSQL.""" diff --git a/kusogaki_bot/data/models.py b/kusogaki_bot/data/models.py index 562f1a6..cf6b885 100644 --- a/kusogaki_bot/data/models.py +++ b/kusogaki_bot/data/models.py @@ -1,7 +1,8 @@ -from db import Base from sqlalchemy import JSON, Column, DateTime, Integer, String from sqlalchemy.sql import func +from kusogaki_bot.data.db import Base + class FoodCounter(Base): __tablename__ = 'food_counters' diff --git a/kusogaki_bot/data/reminder_repository.py b/kusogaki_bot/data/reminder_repository.py index 0af4973..685cc26 100644 --- a/kusogaki_bot/data/reminder_repository.py +++ b/kusogaki_bot/data/reminder_repository.py @@ -1,10 +1,11 @@ import logging from typing import Dict -from db import Database -from models import Reminder from sqlalchemy.exc import SQLAlchemyError +from kusogaki_bot.data.db import Database +from kusogaki_bot.data.models import Reminder + class ReminderRepository: """Repository class for reminder persistence using PostgreSQL.""" diff --git a/kusogaki_bot/data/scheduled_thread_repository.py b/kusogaki_bot/data/scheduled_thread_repository.py index 9b5c181..3ff6501 100644 --- a/kusogaki_bot/data/scheduled_thread_repository.py +++ b/kusogaki_bot/data/scheduled_thread_repository.py @@ -1,10 +1,11 @@ import logging from typing import Dict -from db import Database -from models import ScheduledThread from sqlalchemy.exc import SQLAlchemyError +from kusogaki_bot.data.db import Database +from kusogaki_bot.data.models import ScheduledThread + class ScheduledThreadRepository: """Repository class for scheduled thread persistence using PostgreSQL."""