ref: Move to OO implementation

Heavily inspired by the `youtube-dl` implementation I moved to OO
implementation where now every source type has its own class, making
easy to add new providers.

Also new it has a fallback back, where if no provider is chose it will
try to download with `YoutubeDlProvideBase`.

Add `_TEST` to each class to make it easy to add test to new providers.

1 files changed, 30 insertions, 0 deletions

diff --git a/importer/providers/providerbase.py b/importer/providers/providerbase.py
new file mode 100644
index 0000000..374b9af
--- /dev/null
+++ b/importer/providers/providerbase.py
@@ -0,0 +1,30 @@
+import os
+from typing import List
+
+
+class ProviderBase:
+    paths: List[str]
+    downloaded: bool
+    regex: str
+
+    _TEST = [{
+        "url": "https://i.imgur.com/fXLMjfp.jpg",
+        "paths": ["source_fXLMjfp.jpg"],
+    }]
+
+    def __init__(self, url: str):
+        self.url = url
+        self.paths = []
+        self.downloaded = False
+
+    def download(self):
+        pass
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if self.paths:
+            for path in self.paths:
+                if os.path.exists(path):
+                    os.unlink(path)