Store alternative instance url on mwdb login (#29)

* Store alternative instance url on `mwdb login`

* Update docs

docs/cmdline.rst

@@ -34,6 +34,12 @@ If you want, you can also provide your API key instead of storing your password
 Just copy your API token from `/profile` view and paste into the CLI. All secrets will be stored in `keyring`
 and your username will be saved in `~/.mwdb` file.
 
+.. versionadded:: 3.4.1
+
+If you're using mwdblib to interact with your self-hosted mwdb-core instance, you can now pass the api url using the
+standard --api-url option and it will be automatically saved in the config file.
+
+
 Looking for recent data
 -----------------------
 

src/cli/authenticator.py

@@ -49,12 +49,13 @@ def get_authenticated_mwdb(self, api_url=None):
             mwdb.login(warn=False)
         return mwdb
 
-    def store_login(self, username, password, api_key):
+    def store_login(self, username, password, api_key, api_url=None):
         """
         Sets credentials into user configuration file and keyring
         :param username: Username to store
         :param password: Password to store
         :param api_key: API key to store
+        :param api_url: Alternative API url to store
         """
         if api_key is not None:
             api = APIClient(api_key=api_key)
@@ -65,6 +66,9 @@ def store_login(self, username, password, api_key):
             self.set_config("username", username)
             keyring.set_password("mwdb", username, password)
 
+        if api_url is not None:
+            self.set_config("api_url", api_url)
+
     def reset_login(self):
         """
         Removes credentials from user configuration file and keyring

src/cli/login.py

@@ -23,10 +23,12 @@ def login_command(ctx, username, password, via_api_key, api_key):
             username = click.prompt("Username")
         if password is None:
             password = click.prompt("Password", hide_input=True)
+
+    api_url = ctx.obj.get("api_url", None)
     authenticator = MwdbAuthenticator()
-    authenticator.store_login(username, password, api_key)
+    authenticator.store_login(username, password, api_key, api_url)
     try:
-        mwdb = authenticator.get_authenticated_mwdb(ctx.obj.get("api_url", None))
+        mwdb = authenticator.get_authenticated_mwdb(api_url)
         # todo: Find more appropriate way to check successful authentication
         mwdb.query("", raise_not_found=False)
     except InvalidCredentialsError: