DATA-868 - removeing schema_name from auth call

Author: ashutosh-benni

README.md

@@ -62,8 +62,7 @@ For API Key authentication:
 key = api_key_from_platform
 secret = api_secret_from_platform
 
-# default value for db_schema is production pipeline schema
-cli.login('apikey', secret_id=secret, key_id=key, db_schema=schema_name)
+cli.login('apikey', secret_id=secret, key_id=key)
 ```
 
 ### Proxy Servers and CA Certs
@@ -82,6 +81,15 @@ Similarly, if you need to use your own CA Certificates, set the CURL_CA_BUNDLE e
 CURL_CA_BUNDLE="/path/to/my/certificates"
 ```
 
+### Selecting database schema
+By default, the production pipeline schema will be considered to retrive data from the tenant.
+To select a perticular development pipeline schema you can use following function call:
+
+```
+db_schema = pipeline_id 
+cli.select_db_schema(schema_name=db_schema)
+```
+
 ### Retrieving Data
 
 The typical workflow for getting data from Sight Machine is:

smsdk/Auth/auth.py

@@ -89,7 +89,7 @@ def _auth_basic(self, email=None, password=None):
 
         return success
 
-    def _auth_apikey(self, secret_id, key_id, db_schema=""):
+    def _auth_apikey(self, secret_id, key_id):
         """
         Authenticate by sending an API key.
 
@@ -104,8 +104,6 @@ def _auth_apikey(self, secret_id, key_id, db_schema=""):
         self.session.headers.update({SM_AUTH_HEADER_SECRET_ID: secret_id})
         self.session.headers.update({SM_AUTH_HEADER_SECRET_ID_OLD: secret_id}) #add v0/v1 compat
         self.session.headers.update({SM_AUTH_HEADER_KEY_ID: key_id})
-        if db_schema:
-            self.session.headers.update({X_SM_DB_SCHEMA: db_schema})
 
         if not self.check_auth():
             raise RuntimeError(

smsdk/client.py

@@ -12,7 +12,7 @@
     from pandas.io.json import json_normalize
 
 from smsdk.utils import get_url
-from smsdk.Auth.auth import Authenticator
+from smsdk.Auth.auth import Authenticator, X_SM_DB_SCHEMA
 from smsdk.tool_register import smsdkentities
 from smsdk.client_v0 import ClientV0
 
@@ -115,6 +115,9 @@ def __init__(self, tenant, site_domain="sightmachine.io", protocol = "https"):
         # Setup Authenticator
         self.auth = Authenticator(self)
         self.session = self.auth.session
+    
+    def select_db_schema(self, schema_name):
+        self.session.headers.update({X_SM_DB_SCHEMA:schema_name})
 
     def get_data_v1(self, ename, util_name, normalize=True, *args, **kwargs):
         """