This commit is contained in:
h z
2024-12-02 18:33:39 +00:00
parent 72807d41c1
commit 83c5662fcb
14 changed files with 236 additions and 48 deletions

23
db/__init__.py Normal file
View File

@@ -0,0 +1,23 @@
#db/__init__.py
from contextlib import contextmanager
from models import Base
from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine
import env_provider
engine = create_engine(env_provider.DB_CONNECT_STRING)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
@contextmanager
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()
def create_all():
with engine.begin() as conn:
Base.metadata.create_all(bind=conn)

21
db/models/BackendLog.py Normal file
View File

@@ -0,0 +1,21 @@
#db/models/BackendLog.py
from sqlalchemy import Column, Integer, String, DateTime, Text
from db.models import Base
import datetime
class BackendLog(Base):
__tablename__ = 'backend_log'
id = Column(Integer, primary_key=True, autoincrement=True)
level = Column(String(50), nullable=False)
message = Column(Text, nullable=False)
timestamp = Column(DateTime, nullable=False)
extra = Column(Text, nullable=True)
def to_dict(self):
return {
"id": self.id,
"level": self.level,
"message": self.message,
"timestamp": self.timestamp,
"extra": self.extra,
}

21
db/models/Markdown.py Normal file
View File

@@ -0,0 +1,21 @@
#db/models/Markdown.py
from sqlalchemy import Column, Text, Integer, String, DateTime
from db.models import Base
import datetime
class Markdown(Base):
__tablename__ = 'markdown'
id = Column(Integer, primary_key=True)
title = Column(String(255), nullable=False)
content = Column(Text, nullable=False)
created_at = Column(DateTime, default=datetime.datetime.utcnow)
path = Column(String(255), nullable=False)
def to_dict(self):
return {
'id': self.id,
'title': self.title,
'content': self.content,
'created_at': self.created_at,
'path': self.path,
}

10
db/models/__init__.py Normal file
View File

@@ -0,0 +1,10 @@
#db/models/__init__.py
import pkgutil
import importlib
from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()
package_name = "db.models"
for _, module_name, _ in pkgutil.iter_modules(__path__):
importlib.import_module(f"{package_name}.{module_name}")

9
db/utils.py Normal file
View File

@@ -0,0 +1,9 @@
#db/utils.py
from db import get_db
from db.models.BackendLog import BackendLog
def insert_backend_log(level, message, extra=None):
log_entry = BackendLog(level=level, message=message, extra=extra)
with get_db() as db:
db.add(log_entry)
db.commit()