Skip to content

Commit de24536

Browse files
committed
Updated imports and removed critiquebrainz musicbrainz_db.__init__.py
1 parent 8dca334 commit de24536

File tree

2 files changed

+1
-26
lines changed

2 files changed

+1
-26
lines changed
Original file line numberDiff line numberDiff line change
@@ -1,26 +1 @@
1-
from typing import Optional
2-
from contextlib import contextmanager
3-
from sqlalchemy import create_engine
4-
from sqlalchemy.orm import sessionmaker, scoped_session, Session
5-
from sqlalchemy.pool import NullPool
6-
7-
engine = None
8-
Session: Optional[Session] = None # noqa: F811
91
DEFAULT_CACHE_EXPIRATION = 12 * 60 * 60 # seconds (12 hours)
10-
11-
12-
def init_db_engine(connect_str):
13-
global engine, Session
14-
engine = create_engine(connect_str, poolclass=NullPool)
15-
Session = scoped_session(
16-
sessionmaker(bind=engine)
17-
)
18-
19-
20-
@contextmanager
21-
def mb_session():
22-
session = Session()
23-
try:
24-
yield session
25-
finally:
26-
session.close()

Diff for: import_musicbrainz_row_ids.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from sqlalchemy import text
2+
from brainzutils import musicbrainz_db
23
import critiquebrainz.db.users as db_users
3-
from critiquebrainz.frontend.external import musicbrainz_db
44
from critiquebrainz.frontend import create_app
55
from critiquebrainz import db
66

0 commit comments

Comments
 (0)