rb-commons 0.7.8__py3-none-any.whl → 0.7.9__py3-none-any.whl

rb_commons/orm/enum.py

@@ -4,4 +4,9 @@ class OriginType(str, Enum):
     MANUAL = "MANUAL"
     ETL = "ETL"
     API = "API"
-    DEFAULT = "DEFAULT"
+    DEFAULT = "DEFAULT"
+
+class MediaSource(str, Enum):
+    ROBO = "ROBO"
+    BILLZ = "BILLZ"
+    BITO = "BITO"

rb_commons/utils/media.py

@@ -1,9 +1,17 @@
+from typing import Optional
+
 from pydantic import ValidationError
+from rb_commons.orm.enum import MediaSource
+
 
 class MediaUtils:
+    BILLZ_S3_ENDPOINT_URL = "https://cdn-grocery.billz.ai/billz"
+    BITTO_S3_ENDPOINT_URL = "https://api.bito.uz/upload-api/public/uploads"
+
+    @classmethod
+    def url_builder(cls, key: str, source: Optional[MediaSource] = MediaSource.ROBO):
+        source = MediaSource.ROBO if source is None else source
 
-    @staticmethod
-    def url_builder(key: str):
         try:
             from rb_commons.configs.config import configs
             DIGITALOCEAN_S3_ENDPOINT_URL = configs.DIGITALOCEAN_S3_ENDPOINT_URL
@@ -13,6 +21,11 @@ class MediaUtils:
             DIGITALOCEAN_S3_ENDPOINT_URL = configs.DIGITALOCEAN_S3_ENDPOINT_URL
             DIGITALOCEAN_STORAGE_BUCKET_NAME = configs.DIGITALOCEAN_STORAGE_BUCKET_NAME
 
-        return "{endpoint_url}/{bucket_name}/{key}" \
-            .format(endpoint_url=DIGITALOCEAN_S3_ENDPOINT_URL,
-                    bucket_name=DIGITALOCEAN_STORAGE_BUCKET_NAME, key=key)
+        media_url = f"{DIGITALOCEAN_S3_ENDPOINT_URL}/{DIGITALOCEAN_STORAGE_BUCKET_NAME}/{key}"
+
+        if source == MediaSource.BILLZ:
+            media_url = f"{cls.BILLZ_S3_ENDPOINT_URL}/{key}"
+        elif source == MediaSource.BITO:
+            media_url = f"{cls.BITTO_S3_ENDPOINT_URL}/{key}"
+
+        return media_url

{rb_commons-0.7.8.dist-info → rb_commons-0.7.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.7.8
+Version: 0.7.9
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/RoboSell-organization/rb-commons
 Author: Abdulvoris

{rb_commons-0.7.8.dist-info → rb_commons-0.7.9.dist-info}/RECORD

@@ -12,7 +12,7 @@ rb_commons/http/base_api.py,sha256=ZiSJX5z-1QbL5RuKm30wf-3kbOu4s2plb8q3v0DQUKo,5
 rb_commons/http/consul.py,sha256=Ioq72VD1jGwoC96set7n2SgxN40olzI-myA2lwKkYi4,1864
 rb_commons/http/exceptions.py,sha256=EGRMr1cRgiJ9Q2tkfANbf0c6-zzXf1CD6J3cmCaT_FA,1885
 rb_commons/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/orm/enum.py,sha256=Y0Zew1tE4djDIj4sfswITTZRbcNG8ljKoJy-CGOZVzE,135
+rb_commons/orm/enum.py,sha256=1hrEKtfOKc4DPZ8-2z6Xt4ieHOZJ6K0vl9XGktiZArw,227
 rb_commons/orm/exceptions.py,sha256=1aMctiEwrPjyehoXVX1l6ML5ZOhmDkmBISzlTD5ey1Y,509
 rb_commons/orm/managers.py,sha256=BrpMvOkRoAPKtUV1SfakXMYd73-RhqIaoj1Wmru1gAs,20319
 rb_commons/orm/querysets.py,sha256=Q4iY_TKZItpNnrbCAvlSwyCxJikhaX5qVF3Y2rVgAYQ,1666
@@ -23,8 +23,8 @@ rb_commons/schemes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 rb_commons/schemes/jwt.py,sha256=ZKLJ5D3fcEmEKySjzbxEgUcza4K-oPoHr14_Z0r9Yic,2495
 rb_commons/schemes/pagination.py,sha256=8VZW1wZGJIPR9jEBUgppZUoB4uqP8ORudHkMwvEJSxg,1866
 rb_commons/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rb_commons/utils/media.py,sha256=J2Zi0J28DhcVQVzt-myNNVuzj9Msaetul53VjZtdDdc,820
-rb_commons-0.7.8.dist-info/METADATA,sha256=VFeytR4r8on0mSERmVL4PHcVctvhiomb2zVl7y8h_0g,6570
-rb_commons-0.7.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rb_commons-0.7.8.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
-rb_commons-0.7.8.dist-info/RECORD,,
+rb_commons/utils/media.py,sha256=yfOe3L1ZrylZpP-n-pN49UlP9zg7oAuG0BOq7PT-9Cs,1300
+rb_commons-0.7.9.dist-info/METADATA,sha256=RjUd7-rPdnYlsNjT6ejRUdU1cjQL1K5HtyEUqFYlEqk,6570
+rb_commons-0.7.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rb_commons-0.7.9.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
+rb_commons-0.7.9.dist-info/RECORD,,