first push

authentication/arvan_image/arvan_storage.py

@@ -0,0 +1,169 @@
+# توابع مورد نیاز برای اتصال به ای پی آی استوریج آروان
+# test
+import boto3
+import logging
+from botocore.exceptions import ClientError
+from django.http import HttpResponse
+from PIL import Image
+import io
+import base64
+
+ARVAN_STORAGE_URL = "https://dmstore.s3.ir-thr-at1.arvanstorage.ir/36bba98f-a813-4667-bd60-33aef708bcba.jpg?AWSAccessKeyId=d5739a44-e663-4f43-99f3-13121a62a9e6&Signature=KpBpHBtAS77Y3hHx53g6bmjlGpc%3D&Expires=1651552380"
+
+
+def connect():
+    logging.basicConfig(level=logging.INFO)
+
+    try:
+        s3_resource = boto3.resource(
+            's3',
+            endpoint_url='https://s3.ir-thr-at1.arvanstorage.ir',
+            aws_access_key_id='b0b563b2-bb60-4faf-b09a-660982e70b00',
+            aws_secret_access_key='abdcccaadbd3d897b5432f72bc91048940e012ffa4f308ba0fe16f28e3c80e57'
+        )
+    except Exception as exc:
+        logging.info(exc)
+    return s3_resource
+
+
+def get_bucket_list():
+    s3_resource = connect()
+    li = []
+    try:
+        for bucket in s3_resource.buckets.all():
+            logging.info(f'bucket_name: {bucket.name}')
+            li.append(bucket.name)
+    except ClientError as exc:
+        logging.error(exc)
+    return li[0]
+
+
+# این تابع رو درتمامی قسمت های سیستم میتونیم برای ذخیره عکس بر روی استوریج آروان استفاده کنیم
+
+# def upload_object(image_data, bucket_name, object_name):
+#     resource_connect = connect()
+#     s3_resource = resource_connect
+#     bucket = s3_resource.Bucket(bucket_name)
+#     with open(object_name, "wb") as fh:
+#         fh.write(base64.standard_b64decode(image_data))
+#     with open(object_name, "rb") as fh:
+#         bucket.put_object(
+#             ACL='public-read',
+#             Body=fh,
+#             Key=object_name
+#         )
+
+def upload_object(image_data, bucket_name, object_name):
+    resource_connect = connect()
+    s3_resource = resource_connect
+    bucket = s3_resource.Bucket(bucket_name)
+    buffer = io.BytesIO()
+    imgdata = base64.b64decode(image_data)
+    img = Image.open(io.BytesIO(imgdata))
+    new_img = img.resize((500, 500))  # x, y
+    new_img.save(buffer, format="PNG")
+    img_b64 = base64.b64encode(buffer.getvalue())
+    with open(object_name, "wb") as fh:
+        fh.write(base64.standard_b64decode(img_b64))
+        # base64.standard_b64decode(image_data)
+    with open(object_name, "rb") as fh:
+        bucket.put_object(
+            ACL='public-read',
+            Body=fh,
+            Key=object_name
+        )
+
+
+
+
+def compress_image(image_data, quality):
+    imgdata = base64.b64decode(image_data)
+    img = Image.open(io.BytesIO(imgdata))
+    buffer = io.BytesIO()
+    img.save(buffer, format='JPEG', quality=quality)
+    compressed_data = buffer.getvalue()
+    img_b64 = base64.b64encode(compressed_data)
+    return img_b64
+
+
+def upload_object_resize(image_data, bucket_name, object_name):
+    resource_connect = connect()
+    s3_resource = resource_connect
+    bucket = s3_resource.Bucket(bucket_name)
+    # compressed_img = compress_image(image_data, quality=200)  # تغییر دهید quality به مقدار دلخواه خود
+    decoded_image_data = base64.b64decode(image_data)
+    image_stream = io.BytesIO(decoded_image_data)
+    bucket.put_object(
+        ACL='public-read',
+        Body=image_stream,
+        Key=object_name
+    )
+
+    # with open(object_name, "wb") as fh:
+    #     fh.write(base64.standard_b64decode(compressed_img))
+    #
+    # with open(object_name, "rb") as fh:
+    #     bucket.put_object(
+    #         ACL='public-read',
+    #         Body=fh,
+    #         Key=object_name
+    #     )
+
+
+# def upload_object_resize(image_data, bucket_name, object_name):
+#     resource_connect = connect()
+#     s3_resource = resource_connect
+#     bucket = s3_resource.Bucket(bucket_name)
+#     buffer = io.BytesIO()
+#     imgdata = base64.b64decode(image_data)
+#     img = Image.open(io.BytesIO(imgdata))
+#     img.save(buffer, format="PNG")
+#     img_b64 = base64.b64encode(buffer.getvalue())
+#     with open(object_name, "wb") as fh:
+#         fh.write(base64.standard_b64decode(img_b64))
+#         # base64.standard_b64decode(image_data)
+#     with open(object_name, "rb") as fh:
+#         bucket.put_object(
+#             ACL='public-read',
+#             Body=fh,
+#             Key=object_name
+#         )
+
+def upload_object_less_size(image_data, bucket_name, object_name):
+    resource_connect = connect()
+    s3_resource = resource_connect
+    bucket = s3_resource.Bucket(bucket_name)
+    buffer = io.BytesIO()
+    imgdata = base64.b64decode(image_data)
+    img = Image.open(io.BytesIO(imgdata))
+    # new_img = img.resize((500, 500))  # x, y
+    img.save(buffer, format="PNG")
+    img_b64 = base64.b64encode(buffer.getvalue())
+    with open(object_name, "wb") as fh:
+        fh.write(base64.standard_b64decode(img_b64))
+        # base64.standard_b64decode(image_data)
+    with open(object_name, "rb") as fh:
+        bucket.put_object(
+            ACL='public-read',
+            Body=fh,
+            Key=object_name
+        )
+
+
+def upload_object_for_poultry_science(image_data, bucket_name, object_name):
+    resource_connect = connect()
+    s3_resource = resource_connect
+    bucket = s3_resource.Bucket(bucket_name)
+    buffer = io.BytesIO()
+    imgdata = base64.b64decode(image_data)
+    new_img = Image.open(io.BytesIO(imgdata))
+    new_img.save(buffer, format="JPEG")
+    img_b64 = base64.b64encode(buffer.getvalue())
+    with open(object_name, "wb") as fh:
+        fh.write(base64.standard_b64decode(img_b64))
+    with open(object_name, "rb") as fh:
+        bucket.put_object(
+            ACL='public-read',
+            Body=fh,
+            Key=object_name
+        )