76 lines
2.0 KiB
Python
76 lines
2.0 KiB
Python
import base64
|
|
import io
|
|
import os
|
|
import random
|
|
import string
|
|
|
|
import boto3
|
|
from PIL import Image
|
|
|
|
from ticket.bucket import RASADYAR_ENDPOINT, RASADYAR_BUCKET_NAME, RASADYAR_ACCESS_KEY, RASADYAR_SECRET_KEY
|
|
|
|
ARVAN_TICKET_GALLERY = "https://profileimagedefault.s3.ir-thr-at1.arvanstorage.ir/"
|
|
# ARVAN_TICKET_GALLERY = "https://ticketgallery.s3.ir-thr-at1.arvanstorage.com/"
|
|
ARVAN_User_Image_URL = 'https://profileimagedefault.s3.ir-thr-at1.arvanstorage.ir/'
|
|
|
|
|
|
def send_image_to_server(image):
|
|
ran = ''.join(random.choices(string.ascii_uppercase + string.digits, k=15))
|
|
imgdata = base64.b64decode(image)
|
|
img = Image.open(io.BytesIO(imgdata))
|
|
|
|
img.thumbnail((500, 500))
|
|
|
|
buffer = io.BytesIO()
|
|
img.save(buffer, format="PNG")
|
|
buffer.seek(0)
|
|
|
|
s3_resource = boto3.resource(
|
|
's3',
|
|
endpoint_url=RASADYAR_ENDPOINT,
|
|
aws_access_key_id=RASADYAR_ACCESS_KEY,
|
|
aws_secret_access_key=RASADYAR_SECRET_KEY
|
|
)
|
|
|
|
bucket = s3_resource.Bucket(RASADYAR_BUCKET_NAME)
|
|
bucket.put_object(
|
|
ACL='public-read',
|
|
Body=buffer,
|
|
Key=ran + '.jpg',
|
|
ContentType='image/png'
|
|
)
|
|
|
|
image_url = f"{RASADYAR_ENDPOINT}/{RASADYAR_BUCKET_NAME}/{ran + '.jpg'}"
|
|
|
|
return image_url
|
|
|
|
|
|
def upload_listed_image(req=None, field=None):
|
|
image_list = []
|
|
if req.data[field] != []:
|
|
for item in req.data[field]:
|
|
image_list.append(send_image_to_server(item))
|
|
req.data.pop(field)
|
|
req.data[field] = image_list
|
|
elif req.data[field] == "":
|
|
req.data[field] = "empty"
|
|
return req
|
|
|
|
|
|
def send_image_to_server_for_poultry_science(image, name):
|
|
s3 = boto3.client(
|
|
's3',
|
|
endpoint_url=RASADYAR_ENDPOINT,
|
|
aws_access_key_id=RASADYAR_ACCESS_KEY,
|
|
aws_secret_access_key=RASADYAR_SECRET_KEY
|
|
)
|
|
|
|
s3.upload_fileobj(
|
|
image,
|
|
RASADYAR_BUCKET_NAME,
|
|
name,
|
|
ExtraArgs={'ACL': 'public-read'} # دسترسی عمومی
|
|
)
|
|
|
|
return f"{RASADYAR_ENDPOINT}/{RASADYAR_BUCKET_NAME}/{name}"
|