inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-04-28 19:46:46 +00:00
Code

Adds option to delete existing data when importing records

Browse Source
This commit is contained in:
Oliver Walters 2022-05-02 10:45:26 +10:00
parent 422e22b8e6
commit cfb523adec
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tasks.py
View File

@ -6,6 +6,7 @@ import sys
import pathlib import pathlib
import re import re
try: try:
from invoke import ctask as task from invoke import ctask as task
except: except:
@ -380,8 +381,8 @@ def export_records(c, filename='data.json'):
print("Data export completed") print("Data export completed")
@task(help={'filename': 'Input filename'}, post=[rebuild_models, rebuild_thumbnails]) @task(help={'filename': 'Input filename', 'clear': 'Clear existing data before import'}, post=[rebuild_models, rebuild_thumbnails])
def import_records(c, filename='data.json'): def import_records(c, filename='data.json', clear=False):
""" """
Import database records from a file Import database records from a file
""" """
@ -394,6 +395,9 @@ def import_records(c, filename='data.json'):
print(f"Error: File '{filename}' does not exist") print(f"Error: File '{filename}' does not exist")
sys.exit(1) sys.exit(1)
if clear:
delete_data(c, force=True)
print(f"Importing database records from '{filename}'") print(f"Importing database records from '{filename}'")
# Pre-process the data, to remove any "permissions" specified for a user or group # Pre-process the data, to remove any "permissions" specified for a user or group
@ -432,6 +436,8 @@ def delete_data(c, force=False):
Warning: This will REALLY delete all records in the database!! Warning: This will REALLY delete all records in the database!!
""" """
print(f"Deleting all data from InvenTree database...")
if force: if force:
manage(c, 'flush --noinput') manage(c, 'flush --noinput')
else: else: