Support loading config from source

yzhangcs · yzhangcs · commit 032bc2fcf8fe · 2021-04-27T00:43:09.000+08:00
diff --git a/supar/utils/config.py b/supar/utils/config.py
@@ -1,20 +1,20 @@
 # -*- coding: utf-8 -*-
 
+import argparse
+import os
 from ast import literal_eval
 from configparser import ConfigParser
 
+import supar
+from supar.utils.fn import download
+
 
 class Config(object):
 
-    def __init__(self, conf=None, **kwargs):
+    def __init__(self, **kwargs):
         super(Config, self).__init__()
 
-        config = ConfigParser()
-        config.read(conf or [])
-        self.update({**dict((name, literal_eval(value))
-                            for section in config.sections()
-                            for name, value in config.items(section)),
-                     **kwargs})
+        self.update(kwargs)
 
     def __repr__(self):
         s = line = "-" * 20 + "-+-" + "-" * 30 + "\n"
@@ -28,6 +28,9 @@ def __repr__(self):
     def __getitem__(self, key):
         return getattr(self, key)
 
+    def __contains__(self, key):
+        return hasattr(self, key)
+
     def __getstate__(self):
         return vars(self)
 
@@ -46,8 +49,25 @@ def update(self, kwargs):
         kwargs.update(kwargs.pop('kwargs', dict()))
         for name, value in kwargs.items():
             setattr(self, name, value)
-
         return self
 
+    def get(self, key, default=None):
+        return getattr(self, key) if hasattr(self, key) else default
+
     def pop(self, key, val=None):
         return self.__dict__.pop(key, val)
+
+    @classmethod
+    def load(cls, conf='', unknown=None, **kwargs):
+        config = ConfigParser()
+        config.read(conf if not conf or os.path.exists(conf) else download(supar.CONFIG.get(conf, conf)))
+        config = dict((name, literal_eval(value))
+                      for section in config.sections()
+                      for name, value in config.items(section))
+        if unknown is not None:
+            parser = argparse.ArgumentParser()
+            for name, value in config.items():
+                parser.add_argument('--'+name.replace('_', '-'), type=type(value), default=value)
+            config.update(vars(parser.parse_args(unknown)))
+        config.update(kwargs)
+        return cls(**config)
