def parse_config(): config = configparser.ConfigParser() config.read('config.ini') # Check for required sections. if 'General' not in config: gen_sec_msg = 'Invalid format for config.ini. Missing "[{0}]" section.' fatal_error(gen_sec_msg.format('General')) # Check for required keys. if 'database_name' not in config['General']: gen_key_msg = 'Invalid format for config.ini. Missing key "{0}" under "[{1}]" section.' fatal_error(gen_key_msg.format('database_name', 'General')) if 'base_directory' not in config['General']: gen_key_msg = 'Invalid format for config.ini. Missing key "{0}" under "[{1}]" section.' fatal_error(gen_key_msg.format('base_directory', 'General')) # Check for invalid values if not config['General']['database_name']: gen_val_msg = 'Invalid format for config.ini. Invalid value for "{1}" key under "[{2}]" section.' fatal_error(gen_val_msg.format('database_name', 'General')) return config
def search_file(filepath): print('Searching database for {0}'.format(filepath), end='\n\n') file_info = None try: file_info = get_fileinfo(filepath) except FileNotFoundError: fatal_error('Check file path; Unable to find file at {0}'.format()) except PermissionError: fatal_error('Insufficient permissions to access {0}'.format(filepath)) except BlockingIOError: fatal_error('Unable to access, another process has opened {0}'.format(filepath)) row = check_file_exists_in_database(4, file_info['hashes']['sha1b32']) if row[0] is not '': print('\033[94mFile found! See {0}\033[0m'.format(os.path.join(settings.base_directory, row[0]))) pass else: print('\033[94mFile not found in database.\033[0m')
def search_file(filepath): print('Searching database for {0}'.format(filepath), end='\n\n') file_info = None try: file_info = get_fileinfo(filepath) except FileNotFoundError: fatal_error('Check file path; Unable to find file at {0}'.format()) except PermissionError: fatal_error('Insufficient permissions to access {0}'.format(filepath)) except BlockingIOError: fatal_error('Unable to access, another process has opened {0}'.format( filepath)) row = check_file_exists_in_database(4, file_info['hashes']['sha1b32']) if row[0] is not '': print('\033[94mFile found! See {0}\033[0m'.format( os.path.join(settings.base_directory, row[0]))) pass else: print('\033[94mFile not found in database.\033[0m')