Dynamically load API version

Author: rossjrw

README.md

@@ -5,6 +5,14 @@ Python wrapper around SCUTTLE API.
 
 ## Usage
 
+Create a wrapper for a given wiki and API version:
+
+```python
+from scuttle import scuttle
+
+wiki = scuttle('en', 1)
+```
+
 ## Testing
 
 Install [Pipenv](https://pypi.org/project/pipenv/) and install development

scuttle/versions/__init__.py

@@ -1,3 +1,6 @@
 #!/usr/bin/env python3
 
 """Provides generic methods for accessing version 1 of the API."""
+
+class Api:
+    pass

scuttle/api_1.py renamed to scuttle/versions/v1.py

@@ -3,12 +3,32 @@
 """Provides methods to interface with a given API instance."""
 
 import requests
+import importlib
 
-def scuttle(wiki, api_version=None):
+print(importlib.import_module('.v1', 'scuttle.versions'))
+
+def scuttle(wiki, api_key, api_version=1):
     """Create a new API wrapper for a given wiki and API version.
 
     str `wiki`: The domain of the wiki.
+    str `api_key`: Your SCUTTLE API key.
     int `api_version`: The version of the API to use. If not provided, defaults
-    to latest.
+    to 1.
     """
+    if api_version is None:
+        raise NotImplementedError
+    return ApiWrapper(wiki, api_key, api_version)
+
+class ApiWrapper:
+    def __init__(self, wiki, api_key, api_version):
+        self.domain = wiki
+        self.api_key = api_key
+        self.version = api_version
 
+        try:
+            api_module = importlib.import_module(".v{}".format(self.version),
+                                                 'scuttle.versions')
+        except ModuleNotFoundError:
+            raise ModuleNotFoundError("API version {} does not exist."
+                                      .format(self.version))
+        self.api = api_module.Api()

scuttle/wrapper.py

@@ -4,7 +4,24 @@
 import pytest
 import scuttle
 
-TOKEN = os.environ['SCUTTLE_API_KEY']
+API_KEY = os.environ['SCUTTLE_API_KEY']
 
-def test_something():
-    wiki = scuttle.scuttle('en', 1)
+def test_basic():
+    wiki = scuttle.scuttle('en', None, 1)
+    assert wiki.domain == 'en'
+    assert wiki.version == 1
+    assert isinstance(wiki.api, scuttle.versions.v1.Api)
+
+def test_get_nonexistent_version():
+    with pytest.raises(ModuleNotFoundError) as e:
+        wiki = scuttle.scuttle('en', None, 0)
+    assert str(e.value) == "API version 0 does not exist."
+
+def test_get_latest_version():
+    wiki = scuttle.scuttle('en', None)
+    assert wiki.version == 1
+
+def test_get_wikis():
+    wiki = scuttle.scuttle('en', API_KEY, 1)
+    print(wiki.wikis())
+    assert False