283375/arcaea-offline
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline.git synced 2025-07-01 12:16:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

impr: use Session instead of Engine in write_database of external parsers

Browse Source
This commit is contained in:
283375
2023-08-26 18:53:07 +08:00
parent a2193a7d1b
commit b334108f8c
2 changed files with 8 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/arcaea_offline/external/arcsong/arcsong_db.py vendored
View File

@ -1,7 +1,6 @@
import sqlite3
from typing import List
from sqlalchemy import Engine
from sqlalchemy.orm import Session
from ...models.songs import ChartInfo
@ -29,9 +28,7 @@ class ArcsongDbParser:
return results
def write_database(self, engine: Engine):
with Session(engine) as session:
results = self.parse()
for result in results:
session.merge(result)
session.commit()
def write_database(self, session: Session):
results = self.parse()
for result in results:
session.merge(result)