refactor: remove unused import and init logging only when run as main

rigdenlab · Jul 25, 2020 · 9aa65d4 · 9aa65d4
1 parent a3cbbac
commit 9aa65d4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/scripts/stats.py b/scripts/stats.py
@@ -8,12 +8,9 @@
 __version__ = "2.0"
 
 import logging
-
-logging.basicConfig(level=logging.INFO)
 import os
 import re
 import requests
-import urllib2
 
 LOG = logging.getLogger(__name__)
 BASE_URL = "http://www.rcsb.org/pdb/rest/"
@@ -38,6 +35,8 @@ def update_readme(n_sols):
 
 
 if __name__ == "__main__":
+    logging.basicConfig(level=logging.INFO)
+
     header = {"Content-Type": "application/x-www-form-urlencoded"}
     url = requests.compat.urljoin(BASE_URL, "search")