codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #922 from WithdewHua/fix-db

Browse Source
This commit is contained in:
jxxghp 2023-10-20 06:53:04 +08:00 committed by GitHub
commit efd1733b56
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
app/db/models/__init__.py
View File

@ -1,5 +1,6 @@
from typing import Any, Self, List from typing import Any, Self, List
from sqlalchemy import inspect
from sqlalchemy.orm import as_declarative, declared_attr, Session from sqlalchemy.orm import as_declarative, declared_attr, Session
from app.db import db_update, db_query from app.db import db_update, db_query
@ -25,6 +26,8 @@ class Base:
payload = {k: v for k, v in payload.items() if v is not None} payload = {k: v for k, v in payload.items() if v is not None}
for key, value in payload.items(): for key, value in payload.items():
setattr(self, key, value) setattr(self, key, value)
if inspect(self).detached:
db.add(self)
@classmethod @classmethod
@db_update @db_update