add_rules support + pep8
@@ -3,15 +3,13 @@ import logging
if __name__ == "__main__":
import plac
import sys
- from credentialdigger import download, scan
+ from credentialdigger import add_rules, download, scan
- commands = {'download': download, 'scan': scan}
+ commands = {'add_rules': add_rules, 'download': download, 'scan': scan}
available_comamnds = f'{", ".join(commands)}'
if len(sys.argv) == 1:
- logging.error(
- f'You did not call any command.\n\
- Available commands: {available_comamnds}'
- )
+ logging.error(f'You did not call any command.\n\
+ Available commands: {available_comamnds}')
exit()
command = sys.argv.pop(1)
sys.argv[0] = f'credentialdigger {command}'
