source: memory: Decorate with entry_point

* Added entry_point decorator to MemorySource
* Fixed MemorySource args() and config(). They now use config_get and
  config_set.

Fixes: #78

Signed-off-by: John Andersen <[email protected]>

Author: John Andersen

CHANGELOG.md

@@ -11,6 +11,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Fixed
 - MemoryOperationImplementationNetwork instantiates OperationImplementations
   using their `withconfig()` method.
+- MemorySource now decorated with `entry_point`
+- MemorySource takes arguments correctly via `config_set` and `config_get`
 
 ## [0.2.0] - 2019-05-23
 ### Added

dffml/source/memory.py

@@ -10,6 +10,8 @@
 from ..repo import Repo
 from .source import BaseSourceContext, \
                     BaseSource
+from ..util.cli.arg import Arg
+from ..util.entrypoint import entry_point
 
 class MemorySourceContext(BaseSourceContext):
 
@@ -26,6 +28,7 @@ async def repo(self, src_url: str) -> Repo:
 class MemorySourceConfig(BaseConfig, NamedTuple):
     repos: List[Repo]
 
+@entry_point('memory')
 class MemorySource(BaseSource):
     '''
     Stores repos in a dict in memory
@@ -40,10 +43,14 @@ def __init__(self, config: BaseConfig) -> None:
             self.mem = {repo.src_url: repo for repo in self.config.repos}
 
     @classmethod
-    def args(self) -> Dict[str, Any]:
-        return {}
+    def args(cls, args, *above) -> Dict[str, Arg]:
+        cls.config_set(args, above, 'keys',
+                Arg(type=str,
+                    nargs='+',
+                    default=[]))
+        return args
 
     @classmethod
-    def config(cls, cmd) -> Dict[str, Any]:
-        keys = getattr(cmd, 'keys', [])
-        return MemorySourceConfig(repos=list(map(Repo, keys)))
+    def config(cls, config, *above):
+        return MemorySourceConfig(repos=list(map(Repo,
+                                  cls.config_get(config, above, 'keys'))))