feat: Split and improve contract migration steps

This commit is contained in:
Louis Holbrook
2021-10-20 15:02:36 +00:00
committed by Philip Wafula
parent 8f1afa094d
commit 13fb67d2d8
96 changed files with 2147 additions and 1327 deletions

View File

@@ -0,0 +1,42 @@
# standard imports
import os
import shutil
import sys
def initialize_dirs(user_dir, force_reset=False):
dirs = {}
dirs['old'] = os.path.join(user_dir, 'old')
dirs['new'] = os.path.join(user_dir, 'new')
dirs['meta'] = os.path.join(user_dir, 'meta')
dirs['custom'] = os.path.join(user_dir, 'custom')
dirs['preferences'] = os.path.join(user_dir, 'preferences')
dirs['txs'] = os.path.join(user_dir, 'txs')
dirs['keyfile'] = os.path.join(user_dir, 'keystore')
dirs['custom_new'] = os.path.join(dirs['custom'], 'new')
dirs['custom_meta'] = os.path.join(dirs['custom'], 'meta')
dirs['preferences_meta'] = os.path.join(dirs['preferences'], 'meta')
dirs['preferences_new'] = os.path.join(dirs['preferences'], 'new')
try:
os.stat(dirs['old'])
except FileNotFoundError:
sys.stderr.write('no users to import. please run create_import_users.py first\n')
sys.exit(1)
if force_reset:
for d in dirs.keys():
if d == 'old':
continue
try:
shutil.rmtree(dirs[d])
except FileNotFoundError:
pass
for d in dirs.keys():
if d == 'old':
continue
os.makedirs(dirs[d])
return dirs