api impl
This commit is contained in:
23
db/__init__.py
Normal file
23
db/__init__.py
Normal 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
21
db/models/BackendLog.py
Normal 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
21
db/models/Markdown.py
Normal 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
10
db/models/__init__.py
Normal 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
9
db/utils.py
Normal 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()
|
||||
Reference in New Issue
Block a user