refactor verbose in debug in code

pull/19/head
deadc0de6 4 years ago
parent f0495e6d00
commit 16bf5f99d4

@ -17,10 +17,16 @@ from catcli.logger import Logger
class Catalog: class Catalog:
def __init__(self, path, pickle=False, verbose=False, force=False): def __init__(self, path, pickle=False, debug=False, force=False):
self.path = path # catalog path '''
self.verbose = verbose # verbosity @path: catalog path
self.force = force # force overwrite if exists @pickle: use pickle
@debug: debug mode
@force: force overwrite if exists
'''
self.path = path
self.debug = debug
self.force = force
self.metanode = None self.metanode = None
self.pickle = pickle self.pickle = pickle
@ -60,19 +66,22 @@ class Catalog:
return self._save_pickle(node) return self._save_pickle(node)
return self._save_json(node) return self._save_json(node)
def _debug(self, text):
if not self.debug:
return
Logger.debug(text)
def _save_pickle(self, node): def _save_pickle(self, node):
'''pickle the catalog''' '''pickle the catalog'''
pickle.dump(node, open(self.path, 'wb')) pickle.dump(node, open(self.path, 'wb'))
if self.verbose: self._debug('Catalog saved to pickle \"{}\"'.format(self.path))
Logger.info('Catalog saved to pickle \"{}\"'.format(self.path))
return True return True
def _restore_pickle(self): def _restore_pickle(self):
'''restore the pickled tree''' '''restore the pickled tree'''
root = pickle.load(open(self.path, 'rb')) root = pickle.load(open(self.path, 'rb'))
if self.verbose: m = 'Catalog imported from pickle \"{}\"'.format(self.path)
m = 'Catalog imported from pickle \"{}\"'.format(self.path) self._debug(m)
Logger.info(m)
return root return root
def _save_json(self, node): def _save_json(self, node):
@ -80,14 +89,12 @@ class Catalog:
exp = JsonExporter(indent=2, sort_keys=True) exp = JsonExporter(indent=2, sort_keys=True)
with open(self.path, 'w') as f: with open(self.path, 'w') as f:
exp.write(node, f) exp.write(node, f)
if self.verbose: self._debug('Catalog saved to json \"{}\"'.format(self.path))
Logger.info('Catalog saved to json \"{}\"'.format(self.path))
return True return True
def _restore_json(self, string): def _restore_json(self, string):
'''restore the tree from json''' '''restore the tree from json'''
imp = JsonImporter() imp = JsonImporter()
root = imp.import_(string) root = imp.import_(string)
if self.verbose: self._debug('Catalog imported from json \"{}\"'.format(self.path))
Logger.info('Catalog imported from json \"{}\"'.format(self.path))
return root return root

@ -62,7 +62,7 @@ Options:
-c --hash Calculate md5 hash [default: False]. -c --hash Calculate md5 hash [default: False].
-r --recursive Recursive [default: False]. -r --recursive Recursive [default: False].
-P --parent Ignore stored relpath [default: True]. -P --parent Ignore stored relpath [default: True].
-V --verbose Be verbose [default: False]. -V --erbose Be verbose [default: False].
-v --version Show version. -v --version Show version.
-h --help Show this screen. -h --help Show this screen.
""".format(BANNER, NAME, CATALOGPATH) """.format(BANNER, NAME, CATALOGPATH)
@ -230,10 +230,10 @@ def main():
banner() banner()
# init noder # init noder
noder = Noder(verbose=args['--verbose'], sortsize=args['--sortsize'], noder = Noder(debug=args['--verbose'], sortsize=args['--sortsize'],
arc=args['--archive']) arc=args['--archive'])
# init catalog # init catalog
catalog = Catalog(args['--catalog'], verbose=args['--verbose'], catalog = Catalog(args['--catalog'], debug=args['--verbose'],
force=args['--force']) force=args['--force'])
# init top node # init top node
top = catalog.restore() top = catalog.restore()

@ -36,9 +36,9 @@ class Noder:
TYPE_STORAGE = 'storage' TYPE_STORAGE = 'storage'
TYPE_META = 'meta' TYPE_META = 'meta'
def __init__(self, verbose=False, sortsize=False, arc=False): def __init__(self, debug=False, sortsize=False, arc=False):
self.hash = True self.hash = True
self.verbose = verbose self.debug = debug
self.sortsize = sortsize self.sortsize = sortsize
self.arc = arc self.arc = arc
if self.arc: if self.arc:
@ -352,8 +352,7 @@ class Noder:
script=False, directory=False, script=False, directory=False,
startpath=None, parentfromtree=False): startpath=None, parentfromtree=False):
'''find files based on their names''' '''find files based on their names'''
if self.verbose: self._debug('searching for \"{}\"'.format(key))
Logger.info('searching for \"{}\"'.format(key))
start = root start = root
if startpath: if startpath:
start = self.get_node(root, startpath) start = self.get_node(root, startpath)
@ -390,8 +389,7 @@ class Noder:
############################################################### ###############################################################
def walk(self, root, path, rec=False): def walk(self, root, path, rec=False):
'''walk the tree for ls based on names''' '''walk the tree for ls based on names'''
if self.verbose: self._debug('walking path: \"{}\"'.format(path))
Logger.info('walking path: \"{}\"'.format(path))
r = anytree.resolver.Resolver('name') r = anytree.resolver.Resolver('name')
found = [] found = []
try: try:
@ -482,6 +480,6 @@ class Noder:
return utils.md5sum(path) return utils.md5sum(path)
def _debug(self, string): def _debug(self, string):
if not self.verbose: if not self.debug:
return return
Logger.debug(string) Logger.debug(string)

@ -17,7 +17,7 @@ class TestFind(unittest.TestCase):
def test_find(self): def test_find(self):
# init # init
catalog = Catalog('fake', force=True, verbose=False) catalog = Catalog('fake', force=True, debug=False)
top = catalog._restore_json(get_fakecatalog()) top = catalog._restore_json(get_fakecatalog())
noder = Noder() noder = Noder()

@ -23,7 +23,7 @@ class TestGraph(unittest.TestCase):
gpath = tempfile.gettempdir() + os.sep + 'graph.dot' gpath = tempfile.gettempdir() + os.sep + 'graph.dot'
self.addCleanup(clean, path) self.addCleanup(clean, path)
self.addCleanup(clean, gpath) self.addCleanup(clean, gpath)
catalog = Catalog(path, force=True, verbose=False) catalog = Catalog(path, force=True, debug=False)
top = catalog._restore_json(get_fakecatalog()) top = catalog._restore_json(get_fakecatalog())
noder = Noder() noder = Noder()

@ -42,7 +42,7 @@ class TestIndexing(unittest.TestCase):
noder = Noder() noder = Noder()
top = noder.new_top_node() top = noder.new_top_node()
catalog = Catalog(catalogpath, force=True, verbose=False) catalog = Catalog(catalogpath, force=True, debug=False)
# create fake args # create fake args
tmpdirname = 'tmpdir' tmpdirname = 'tmpdir'

@ -19,7 +19,7 @@ class TestWalking(unittest.TestCase):
# init # init
path = 'fake' path = 'fake'
self.addCleanup(clean, path) self.addCleanup(clean, path)
catalog = Catalog(path, force=True, verbose=False) catalog = Catalog(path, force=True, debug=False)
top = catalog._restore_json(get_fakecatalog()) top = catalog._restore_json(get_fakecatalog())
noder = Noder() noder = Noder()

@ -19,7 +19,7 @@ class TestRm(unittest.TestCase):
# init # init
path = 'fake' path = 'fake'
self.addCleanup(clean, path) self.addCleanup(clean, path)
catalog = Catalog(path, force=True, verbose=False) catalog = Catalog(path, force=True, debug=False)
top = catalog._restore_json(get_fakecatalog()) top = catalog._restore_json(get_fakecatalog())
noder = Noder() noder = Noder()

@ -19,7 +19,7 @@ class TestTree(unittest.TestCase):
# init # init
path = 'fake' path = 'fake'
self.addCleanup(clean, path) self.addCleanup(clean, path)
catalog = Catalog(path, force=True, verbose=False) catalog = Catalog(path, force=True, debug=False)
top = catalog._restore_json(get_fakecatalog()) top = catalog._restore_json(get_fakecatalog())
noder = Noder() noder = Noder()

@ -44,7 +44,7 @@ class TestIndexing(unittest.TestCase):
noder = Noder() noder = Noder()
top = noder.new_top_node() top = noder.new_top_node()
catalog = Catalog(catalogpath, force=True, verbose=False) catalog = Catalog(catalogpath, force=True, debug=False)
# get checksums # get checksums
f4_md5 = md5sum(f4) f4_md5 = md5sum(f4)

Loading…
Cancel
Save