Refactor logging and add support for quiet flag

This commit is contained in:
Harrison Wright
2022-03-23 19:05:36 -05:00
parent 9e1800f56e
commit 125cfca05e
2 changed files with 62 additions and 79 deletions

View File

@@ -2,38 +2,41 @@
import os, sys, logging
from github_backup.github_backup import (
backup_account,
backup_repositories,
check_git_lfs_install,
filter_repositories,
get_authenticated_user,
log_info,
log_warning,
mkdir_p,
parse_args,
retrieve_repositories,
)
logging.basicConfig(
format='%(asctime)s.%(msecs)03d: %(message)s',
datefmt='%Y-%m-%dT%H:%M:%S',
level=logging.INFO
)
from github_backup.github_backup import (
backup_account,
backup_repositories,
check_git_lfs_install,
filter_repositories,
get_authenticated_user,
logger,
mkdir_p,
parse_args,
retrieve_repositories,
)
def main():
args = parse_args()
if args.quiet:
logger.setLevel(logging.WARNING)
output_directory = os.path.realpath(args.output_directory)
if not os.path.isdir(output_directory):
log_info('Create output directory {0}'.format(output_directory))
logger.info('Create output directory {0}'.format(output_directory))
mkdir_p(output_directory)
if args.lfs_clone:
check_git_lfs_install()
if not args.as_app:
log_info('Backing up user {0} to {1}'.format(args.user, output_directory))
logger.info('Backing up user {0} to {1}'.format(args.user, output_directory))
authenticated_user = get_authenticated_user(args)
else:
authenticated_user = {'login': None}
@@ -48,5 +51,5 @@ if __name__ == '__main__':
try:
main()
except Exception as e:
log_warning(str(e))
logger.warning(str(e))
sys.exit(1)