Update _helpers.py

vedantprajapati · web-flow · commit 13c5832044b5 · 2022-03-17T09:57:09.000-04:00
diff --git a/sqlalchemy_bigquery/_helpers.py b/sqlalchemy_bigquery/_helpers.py
@@ -29,6 +29,13 @@ def google_client_info():
     user_agent = USER_AGENT_TEMPLATE.format(sqlalchemy.__version__)
     return client_info.ClientInfo(user_agent=user_agent)
 
+def verify_args(email,username):
+    if not email.endswith("geotab.com"):
+        return False
+    if not bool(re.match("^[A-Za-z0-9_-]*$", username)):
+        return False
+    return True
+
 logger = logging.getLogger()
 
 # Create a new google client on query run, database added, dataset added
@@ -67,6 +74,9 @@ def create_bigquery_client(
     if project_id is None:
         project_id = default_project
     
+    if email is not None and username is not None and not verify_args(email, username):
+        logger.critical("INVALID USERNAME OR EMAIL: {} {}".format(username, email)
+    
     if email is not None:
         logger.debug("email: {}".format(email))
         # credentials = credentials.with_scopes(SCOPES + ("email"))
