Merge pull request #22 from thoughtspot/v.1.6.1

V.1.6.1

Author: bryanthowell-ts

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = thoughtspot_rest_api_v1
-version = 1.6.0
+version = 1.6.1
 description = Library implementing the ThoughtSpot V1 REST API
 long_description = file: README.md
 long_description_content_type = text/markdown

src/thoughtspot_rest_api_v1/_version.py

@@ -1 +1 @@
-__version__ = '1.6.0'
+__version__ = '1.6.1'

src/thoughtspot_rest_api_v1/tsrestapiv2.py

@@ -82,7 +82,8 @@ def bearer_token(self, bearer_token):
     #
     def auth_session_login(self,  username: Optional[str] = None, password: Optional[str] = None,
                            remember_me: bool = True,
-                           bearer_token: Optional[str] = None) -> requests.Session:
+                           bearer_token: Optional[str] = None,
+                           org_identifier: Optional[int] = None) -> requests.Session:
         endpoint = 'auth/session/login'
 
         url = self.base_url + endpoint
@@ -97,6 +98,8 @@ def auth_session_login(self,  username: Optional[str] = None, password: Optional
                 'password': password,
                 'remember_me': str(remember_me).lower()
             }
+            if org_identifier is not None:
+                json_post_data["org_identifier"] = org_identifier
             response = self.requests_session.post(url=url, json=json_post_data)
         else:
             raise Exception("If using username/password, must include both")
@@ -367,8 +370,6 @@ def orgs_delete(self, org_identifier: str):
     #
     def tags_search(self, tag_identifier: Optional[str] = None, color: Optional[str] = None):
         endpoint = 'tags/search'
-        if tag_identifier is None and color is None:
-            raise Exception("Must provide tag_identifier or color")
         request = {}
         if tag_identifier is not None:
             request['tag_identifier'] = tag_identifier
@@ -471,8 +472,9 @@ def metadata_tml_export(self, metadata_ids: List[str], export_associated: bool =
             'export_fqn': export_fqn
         }
         # These are left as optionals / defaults because they may have been added after 9.5
-        if edoc_format.upper() == 'YAML':
-            request['edoc_format'] = 'YAML'
+        if edoc_format is not None:
+            if edoc_format.upper() == 'YAML':
+                request['edoc_format'] = 'YAML'
         if export_schema_version is not None:
             request['export_schema_version'] = export_schema_version
         if metadata_request is not None: