Spaces:
Running
Running
File size: 3,812 Bytes
4051191 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import os
from typing import Type, List, TypeVar, Optional
from sqlalchemy.ext.asyncio import create_async_engine
from sqlmodel import SQLModel, Field, Session, select, delete
from sqlmodel.ext.asyncio.session import AsyncSession
from tools import LanguageSingleton
T = TypeVar("T")
class ChapterFile(SQLModel, table=True):
url: str = Field(primary_key=True)
file_id: Optional[str]
file_unique_id: Optional[str]
cbz_id: Optional[str]
cbz_unique_id: Optional[str]
telegraph_url: Optional[str]
class MangaOutput(SQLModel, table=True):
user_id: str = Field(primary_key=True, regex=r'\d+')
output: int = Field
class Subscription(SQLModel, table=True):
url: str = Field(primary_key=True)
user_id: str = Field(primary_key=True, regex=r'\d+')
class LastChapter(SQLModel, table=True):
url: str = Field(primary_key=True)
chapter_url: str = Field
class MangaName(SQLModel, table=True):
url: str = Field(primary_key=True)
name: str = Field
class DB(metaclass=LanguageSingleton):
def __init__(self, dbname: str = 'sqlite+aiosqlite:///test.db'):
if dbname.startswith('postgres://'):
dbname = dbname.replace('postgres://', 'postgresql+asyncpg://', 1)
elif dbname.startswith('postgresql://'):
dbname = dbname.replace('postgresql://', 'postgresql+asyncpg://', 1)
elif dbname.startswith('sqlite'):
dbname = dbname.replace('sqlite', 'sqlite+aiosqlite', 1)
self.engine = create_async_engine(dbname)
async def connect(self):
async with self.engine.begin() as conn:
await conn.run_sync(SQLModel.metadata.create_all, checkfirst=True)
async def add(self, other: SQLModel):
async with AsyncSession(self.engine) as session: # type: AsyncSession
async with session.begin():
session.add(other)
async def get(self, table: Type[T], id) -> T:
async with AsyncSession(self.engine) as session: # type: AsyncSession
return await session.get(table, id)
async def get_all(self, table: Type[T]) -> List[T]:
async with AsyncSession(self.engine) as session: # type: AsyncSession
statement = select(table)
return await session.exec(statement=statement)
async def erase(self, other: SQLModel):
async with AsyncSession(self.engine) as session: # type: AsyncSession
async with session.begin():
await session.delete(other)
async def get_chapter_file_by_id(self, id: str):
async with AsyncSession(self.engine) as session: # type: AsyncSession
statement = select(ChapterFile).where((ChapterFile.file_unique_id == id) |
(ChapterFile.cbz_unique_id == id) |
(ChapterFile.telegraph_url == id))
return (await session.exec(statement=statement)).first()
async def get_subs(self, user_id: str, filters=None) -> List[MangaName]:
async with AsyncSession(self.engine) as session:
statement = (
select(MangaName)
.join(Subscription, Subscription.url == MangaName.url)
.where(Subscription.user_id == user_id)
)
for filter_ in filters or []:
statement = statement.where(MangaName.name.ilike(f'%{filter_}%') | MangaName.url.ilike(f'%{filter_}%'))
return (await session.exec(statement=statement)).all()
async def erase_subs(self, user_id: str):
async with AsyncSession(self.engine) as session:
async with session.begin():
statement = delete(Subscription).where(Subscription.user_id == user_id)
await session.exec(statement=statement)
|