42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
|
# 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['phone'] = os.path.join(user_dir, 'phone')
|
||
|
|
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['phone_meta'] = os.path.join(dirs['phone'], 'meta')
|
||
|
|
|
||
|
|
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
|