From 1104dde2d7d7b189b1f7810e03278a82ebc92bde Mon Sep 17 00:00:00 2001 From: Samuel Grayson Date: Tue, 15 Mar 2022 15:29:23 -0500 Subject: [PATCH] Fix __main__ for pytest --- ascl_net_scraper/__main__.py | 36 ++++++++++++++++++++---------------- 1 file changed, 20 insertions(+), 16 deletions(-) diff --git a/ascl_net_scraper/__main__.py b/ascl_net_scraper/__main__.py index 5bd5473..652d432 100644 --- a/ascl_net_scraper/__main__.py +++ b/ascl_net_scraper/__main__.py @@ -1,24 +1,28 @@ import logging +import os from tqdm import tqdm from .main import scrape_index_list, scrape_details -logger = logging.getLogger("charmonium.freeze") -logger.setLevel(logging.DEBUG) -fh = logging.FileHandler("freeze.log") -fh.setLevel(logging.DEBUG) -fh.setFormatter(logging.Formatter("%(asctime)s: %(message)s")) -logger.addHandler(fh) +# __name__ == "__main__" is needed so pytest ignores this. +if __name__ == "__main__": + logger = logging.getLogger("charmonium.freeze") + logger.setLevel(logging.DEBUG) + fh = logging.FileHandler("freeze.log") + fh.setLevel(logging.DEBUG) + fh.setFormatter(logging.Formatter("%(message)s")) + logger.addHandler(fh) + logger.debug("Program {}", os.getpid()) -logger = logging.getLogger("charmonium.cache.ops") -logger.setLevel(logging.DEBUG) -fh = logging.FileHandler("cache.log") -fh.setLevel(logging.DEBUG) -fh.setFormatter(logging.Formatter("%(asctime)s: %(message)s")) -logger.addHandler(fh) + logger = logging.getLogger("charmonium.cache.ops") + logger.setLevel(logging.DEBUG) + fh = logging.FileHandler("cache.log") + fh.setLevel(logging.DEBUG) + fh.setFormatter(logging.Formatter("%(message)s")) + logger.addHandler(fh) + logger.debug("Program {}", os.getpid()) - -records = scrape_index_list(10) -for record in tqdm(records, total=len(records)): - detailed_record = scrape_details(record.details_url) + records = scrape_index_list(10) + for record in tqdm(records, total=len(records)): + detailed_record = scrape_details(record.details_url)