rdxz2-utill 0.0.8__py3-none-any.whl → 0.0.10__py3-none-any.whl

{rdxz2_utill-0.0.8.dist-info → rdxz2_utill-0.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rdxz2-utill
-Version: 0.0.8
+Version: 0.0.10
 Summary: Your daily Python utility
 Author-email: Richard Dharmawan <richard.dharmawan@gmail.com>
 License: MIT License

{rdxz2_utill-0.0.8.dist-info → rdxz2_utill-0.0.10.dist-info}/RECORD

@@ -1,6 +1,6 @@
-rdxz2_utill-0.0.8.dist-info/licenses/LICENSE,sha256=PF9CUvzP8XFYopEAzrMzSCovF7RdBdscPqJCDC6KjPc,1073
+rdxz2_utill-0.0.10.dist-info/licenses/LICENSE,sha256=PF9CUvzP8XFYopEAzrMzSCovF7RdBdscPqJCDC6KjPc,1073
 utill/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-utill/my_bq.py,sha256=V2MVVciiAR8n840Cr7hFeTuisCPEQDyhjLMM6CCwQFE,14676
+utill/my_bq.py,sha256=zxmQJUojJCHSZ7fo20ybbwz_cXnbCh9XNw4feKQJcZc,14848
 utill/my_compare.py,sha256=619QbVk3GihWxen95yVnivKHkah8GgPTLGiSkgHxykw,886
 utill/my_const.py,sha256=88dOqn6NPQ5-hfRqdkew5POoAIyO91XXOGvN76oNsdo,251
 utill/my_csv.py,sha256=svgu93R0pP7UW0B58eJMi0vuJnYhqMtafzCsTIk4yUU,2781
@@ -9,10 +9,10 @@ utill/my_dict.py,sha256=jPaPfdn4WYpm0uIBPiYFinpHhx1jXpFVDJ9npmvxGZQ,391
 utill/my_encryption.py,sha256=SCF7PPur39cW4RHidsRhw-9BZP-ymUH-6LZ9nAHJDsY,2105
 utill/my_env.py,sha256=mREys72Ybg2p9p2s7ApOt0s_6F5-qxR8FyYEcSJ8pmU,2093
 utill/my_file.py,sha256=H3QmIOwubQCUMoOuk7jwf6AnqsljWZIuM7OjelyZby4,1865
-utill/my_gcs.py,sha256=u9rZr4e6XLLFK1QC9JknMwBeNX5MWOnVq1W50U5Jt14,4032
+utill/my_gcs.py,sha256=KUx89rZx2-dq-GV1LbbvbZ79Qr9NznjG1Zipop4hMZE,4216
 utill/my_input.py,sha256=OyKLoutXpwISReltuL_Gw2oojv16tYWJqQpqabBOQx4,350
 utill/my_json.py,sha256=WgW6mavGhfs4h1N5XbhsDnRk2dbh_ttJWdJUj4iWDN4,1473
-utill/my_mb.py,sha256=3_A5kXHgnkxGbd38vK5t5MfFcj84lohjS7C2OtlSo30,14841
+utill/my_mb.py,sha256=IyrySs92TqtjBUvPMeUN3P2kRK8EttTFRPZsv5Cr-xw,15090
 utill/my_pg.py,sha256=ltNPAsrOJurcCQTDZfAdsOXBSnbmxPzu4E-VtMvUxcM,6708
 utill/my_queue.py,sha256=hINP4_yjmboSjHgo1J3CtPm2X9SE3HfczyED3ip7nfk,1930
 utill/my_string.py,sha256=pINYFR1ligTyVZYzV8P_FolCsZQwYE1jaFNTuQ3XS_8,833
@@ -28,8 +28,8 @@ utill/cmd/_pg.py,sha256=RVxEiSifyIwMDYDM69vt6WSLdVDr1cMzY6r4T2PzNRA,492
 utill/cmd/utill.py,sha256=TlHfiwOUcK1m58PrRCjX9sARiPYZUsoTk-KOTCOz1vM,3558
 utill/templates/mb.json,sha256=M46ZHSaSh4rbD_KGUViGr2B2ZV8_PC-O5Evqi35JK5g,59
 utill/templates/pg.json,sha256=LkJt0VV3zcyt7Tpn6gulsoVQgUc-9uImXOStvzu8cdU,271
-rdxz2_utill-0.0.8.dist-info/METADATA,sha256=pS9jZSY-KHkjJ9SBvewLcnMkvCVmXZye0N9qzv7veSc,4401
-rdxz2_utill-0.0.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rdxz2_utill-0.0.8.dist-info/entry_points.txt,sha256=9n5NWz5Wi9jDvYhB_81_4icgT5xABZ-QivHD8ibcafg,47
-rdxz2_utill-0.0.8.dist-info/top_level.txt,sha256=tuAYZoCsr02JYbpZj7I6fl1IIo53v3GG0uoj-_fINVk,6
-rdxz2_utill-0.0.8.dist-info/RECORD,,
+rdxz2_utill-0.0.10.dist-info/METADATA,sha256=XNWgh3Y8eF9C7yX_VGUSTtoMB0Cc9A9m9x4Q-1g2PLk,4402
+rdxz2_utill-0.0.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rdxz2_utill-0.0.10.dist-info/entry_points.txt,sha256=9n5NWz5Wi9jDvYhB_81_4icgT5xABZ-QivHD8ibcafg,47
+rdxz2_utill-0.0.10.dist-info/top_level.txt,sha256=tuAYZoCsr02JYbpZj7I6fl1IIo53v3GG0uoj-_fINVk,6
+rdxz2_utill-0.0.10.dist-info/RECORD,,

utill/my_bq.py

@@ -58,11 +58,15 @@ class Dtype:
 
 
 class BQ():
-    def __init__(self, project: str = None):
+    def __init__(self, project: str = None, service_account_filename: str = None):
         self.project = project or envs.GCP_PROJECT_ID
 
-        self.client = bigquery.Client(project=self.project)
-        logger.debug(f'BQ client open, project: {self.project or "<application-default>"}')
+        if service_account_filename is not None:
+            self.client = bigquery.Client.from_service_account_json(service_account_filename)
+        else:
+            self.client = bigquery.Client(project=self.project)
+
+        logger.debug(f'BQ client open, project: {self.client.project}')
 
     def __enter__(self):
         return self
@@ -247,7 +251,7 @@ class BQ():
     def download_csv(self, query: str, dst_filename: str, combine: bool = True, pre_query: str = None):
         if not dst_filename.endswith('.csv'):
             raise ValueError('Destination filename must ends with .csv!')
-        
+
         dst_filename = os.path.expanduser(dst_filename)
 
         dirname = dst_filename.removesuffix('.csv')

utill/my_gcs.py

@@ -9,16 +9,20 @@ from .my_env import envs
 
 class GCS:
 
-    def __init__(self, project: str = None, bucket_name: str = None):
+    def __init__(self, project: str = None, service_account_filename: str = None, bucket_name: str = None):
         self.project = project if project is not None else envs.GCP_PROJECT_ID
-        self.client = storage.Client(project=self.project)
+
+        if service_account_filename is not None:
+            self.client = storage.Client.from_service_account_json(service_account_filename)
+        else:
+            self.client = storage.Client(project=self.project)
 
         bucket_name_parts = (bucket_name or envs.GCS_BUCKET).split('/')
         self.change_bucket(bucket_name_parts[0])
         self.base_path = '/'.join(bucket_name_parts[1:]) if len(bucket_name_parts) > 1 else None
         not self.base_path or logger.debug(f'Base path: {self.base_path}')
 
-        logger.debug(f'GCS client open, project: {project or "<application-default>"}')
+        logger.debug(f'GCS client open, project: {self.client.project}')
 
     def __enter__(self):
         return self

utill/my_mb.py

@@ -12,6 +12,12 @@ from .my_dict import AutoPopulatingDict
 from .my_env import MB_FILENAME
 
 
+class UsersNotFoundException(Exception):
+    def __init__(self, username: list[str]):
+        self.message = f'User(s) {username} not exists'
+        super().__init__(self.message)
+
+
 def _decode_collection_location_to_group(collections_dict: dict, location: str):
     return ' > '.join(map(lambda x: collections_dict[x], map(int, location.strip('/').split('/'))))
 
@@ -21,10 +27,10 @@ def _translate_user_group_ids(user: dict) -> set:
 
 
 class MB:
-    def __init__(self, base_url: str = None) -> None:
-        config = json.loads(open(MB_FILENAME, 'r').read())
+    def __init__(self, config_source: str = MB_FILENAME) -> None:
+        config = json.loads(open(config_source, 'r').read())
 
-        self.base_url = base_url or config['base_url']
+        self.base_url = config['base_url']
         self.api_key = config['api_key']
 
         self._is_user_initialized = False
@@ -98,7 +104,7 @@ class MB:
                 not_exists.append(email)
 
         if not_exists:
-            raise ValueError(f'Email not exists: {not_exists}')
+            raise UsersNotFoundException(not_exists)
 
     def create_user(self, first_name: str, last_name: str, email: str, group_ids: list):
         self.send_request(HttpMethod.POST, 'api/user', {
@@ -111,7 +117,11 @@ class MB:
         logger.info(f'✅ Create user {email}')
 
     def deactivate_user_by_email(self, email: str):
-        user = self.dict__user_email__user[email]
+        try:
+            user = self.dict__user_email__user[email]
+        except KeyError as e:
+            raise UsersNotFoundException([email])
+
         self.send_request(HttpMethod.DELETE, f'api/user/{user["id"]}')
         del self.dict__user_email__user[email]
         logger.info(f'✅ Deactivate user [{user["id"]}] {email}')
@@ -120,8 +130,7 @@ class MB:
         try:
             self.dict__user_email__user[email]
         except KeyError as e:
-            logger.error(f'User {email} not exists')
-            raise e
+            raise UsersNotFoundException([email])
         self.send_request(HttpMethod.POST, 'api/session/forgot_password', {'email': email})
         logger.info(f'✅ Reset password {email}')