zeroGRMh
f53783efd3
- Added test file to gitignore
- Created table of users with chat_ids(UserTable.sql)
- Added check of empty table into get_last_id
- Modified get_user to return a chat_id instead of username
- Changed download function in ISwork to send URL instead
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
from minio import Minio
|
|
from random import randint
|
|
from datetime import timedelta
|
|
from src.Backend import DBwork
|
|
from src import config
|
|
|
|
|
|
def _setClient():
|
|
minio_client = Minio(
|
|
config.IS_address,
|
|
access_key = config.acc_key,
|
|
secret_key = config.sec_key,
|
|
secure = False
|
|
)
|
|
return minio_client
|
|
|
|
def getNumberofObjects(client, currentDay):
|
|
objects = client.list_objects(config.bucket_name, prefix=currentDay+'/')
|
|
return sum(1 for _ in objects)
|
|
|
|
def getObjectExtension(client, currentDay, fileNumber):
|
|
objects = client.list_objects(config.bucket_name, prefix=currentDay+'/')
|
|
for counter, obj in enumerate(objects, start=1):
|
|
if counter == fileNumber:
|
|
return obj.object_name.split('.')[-1]
|
|
|
|
def getImageName(currentDay, client):
|
|
maxFiles = getNumberofObjects(client, currentDay)
|
|
fileNumber = randint(1, maxFiles)
|
|
fileExtension = '.' + getObjectExtension(client, currentDay, fileNumber)
|
|
desiredFile = currentDay + '/' + str(fileNumber) + fileExtension
|
|
return desiredFile
|
|
|
|
def getDownloadURL(currentDay):
|
|
client = _setClient()
|
|
object_name = getImageName(currentDay, client)
|
|
url = client.presigned_get_object(
|
|
config.bucket_name,
|
|
object_name,
|
|
expires=timedelta(days=1))
|
|
return url
|
|
|
|
def downloadForAll(currentDay):
|
|
cur, conn = DBwork.set_connection()
|
|
max_id = DBwork.get_last_id(cur)
|
|
for id in range(1, max_id + 1):
|
|
chat_id = DBwork.get_chat_id(id, cur)
|
|
image_URL = getDownloadURL(currentDay)
|
|
DBwork.close_connection(conn, cur)
|
|
|