mirror of
https://github.com/283375/arcaea-offline.git
synced 2025-04-17 21:30:18 +00:00
impr: use Session
instead of Engine
in write_database
of external parsers
This commit is contained in:
parent
a2193a7d1b
commit
b334108f8c
11
src/arcaea_offline/external/arcaea/common.py
vendored
11
src/arcaea_offline/external/arcaea/common.py
vendored
@ -2,7 +2,6 @@ import json
|
||||
from os import PathLike
|
||||
from typing import Any, List, Optional, Union
|
||||
|
||||
from sqlalchemy import Engine
|
||||
from sqlalchemy.orm import DeclarativeBase, Session
|
||||
|
||||
|
||||
@ -47,9 +46,7 @@ class ArcaeaParser:
|
||||
def parse(self) -> List[DeclarativeBase]:
|
||||
...
|
||||
|
||||
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)
|
||||
|
@ -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)
|
||||
|
Loading…
x
Reference in New Issue
Block a user