diff --git a/db/data.py b/db/data.py index 9baf4c311..5e468ec0d 100644 --- a/db/data.py +++ b/db/data.py @@ -10,6 +10,7 @@ from sqlalchemy import text import sqlalchemy.exc +from brainzutils import musicbrainz_db _whitelist_file = os.path.join(os.path.dirname(__file__), "tagwhitelist.json") _whitelist_tags = set(json.load(open(_whitelist_file))) diff --git a/manage.py b/manage.py index 4236998af..81788178c 100644 --- a/manage.py +++ b/manage.py @@ -202,7 +202,7 @@ def get_entities(): print('Redirecting mbids to original entities...') webserver.external.get_entities.get_original_entity() - +@cli.command() def apply_replication_changes(): print("\nUpdating musicbrainz schema by applying replication packets...") musicbrainz_importer.apply_replication_changes.main() diff --git a/musicbrainz_importer/apply_replication_changes.py b/musicbrainz_importer/apply_replication_changes.py index 1aed3c850..f7f4d8c09 100644 --- a/musicbrainz_importer/apply_replication_changes.py +++ b/musicbrainz_importer/apply_replication_changes.py @@ -36,6 +36,7 @@ from sqlalchemy import text from sqlalchemy.exc import IntegrityError import db.import_mb_data +import db.data include_tables = ['language', 'artist_credit_name', 'artist', 'artist_gid_redirect', 'area', 'area_type', 'recording_gid_redirect', \ 'script', 'release_gid_redirect', 'recording', 'track', 'artist_credit', 'release_group_primary_type', 'release_group', \ @@ -298,6 +299,7 @@ def process_tar(fileobj, expected_schema_seq, replication_seq): elif member.name in ('mbdump/PendingData', 'mbdump/dbmirror_pendingdata'): importer.load_pending_data(tar.extractfile(member)) importer.process() + tar.close() def download_packet(base_url, token, replication_seq):