github/nzbToMedia
1
0
Fork 0
mirror of https://github.com/clinton-hall/nzbToMedia.git synced 2025-08-19 12:59:36 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Hotfix cleaning for source installs

Browse source
This commit is contained in:
Labrys of Knossos 2019-01-05 21:26:56 -05:00
commit d250e45c7b
1 changed files with 41 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

45
cleanup.py
View file

@ -5,6 +5,20 @@ from __future__ import print_function
import os import os
import subprocess import subprocess
import sys import sys
import shutil
FOLDER_STRUCTURE = {
'libs': [
'common',
'custom',
'py2',
'win',
],
'core': [
'auto_process',
'extractor',
],
}
class WorkingDirectory(object): class WorkingDirectory(object):
@ -106,7 +120,22 @@ def clean_folders(*paths):
return result return result
def clean(*paths): def force_clean_folder(path, required):
root, dirs, files = next(os.walk(path))
required = sorted(required)
if required:
print('Skipping required subfolders', required)
remove = sorted(set(dirs).difference(required))
missing = sorted(set(required).difference(dirs))
for path in remove:
pathname = os.path.join(root, path)
print('Removing', pathname)
shutil.rmtree(pathname)
if missing:
raise Exception('Required subfolders missing:', missing)
def clean(paths):
"""Clean up bytecode and obsolete folders.""" """Clean up bytecode and obsolete folders."""
with WorkingDirectory(module_path()) as cwd: with WorkingDirectory(module_path()) as cwd:
if cwd.working_directory != cwd.original_directory: if cwd.working_directory != cwd.original_directory:
@ -121,7 +150,7 @@ def clean(*paths):
print(result or 'No bytecode to clean') print(result or 'No bytecode to clean')
if paths and os.path.exists('.git'): if paths and os.path.exists('.git'):
print('\n-- Cleaning folders: {} --'.format(paths)) print('\n-- Cleaning folders: {} --'.format(list(paths)))
try: try:
result = clean_folders(*paths) result = clean_folders(*paths)
except SystemExit as error: except SystemExit as error:
@ -129,7 +158,15 @@ def clean(*paths):
else: else:
print(result or 'No folders to clean\n') print(result or 'No folders to clean\n')
else: else:
print('Directory is not a git repository') print('\nDirectory is not a git repository')
try:
items = paths.items()
except AttributeError:
print('Failed to clean, no subfolder structure given')
else:
for folder, subfolders in items:
print('\nForce cleaning folder:', folder)
force_clean_folder(folder, subfolders)
if cwd.working_directory != cwd.original_directory: if cwd.working_directory != cwd.original_directory:
print('Returning to directory: ', cwd.original_directory) print('Returning to directory: ', cwd.original_directory)
@ -138,4 +175,4 @@ def clean(*paths):
if __name__ == '__main__': if __name__ == '__main__':
clean('libs', 'core') clean(FOLDER_STRUCTURE)