n0one/CatBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

deleted dependency on '.jpeg' extension

Browse Source
This commit is contained in:
zeroGRMh
2025-07-28 15:25:42 +03:00
parent 2529138c4e
commit 79a2f1d7e7
1 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/Backend/ISwork.py
View File

@ -13,19 +13,28 @@ def _setClient():
) )
return minio_client return minio_client
def getNumberofObjects(client, currentDay, bucket_name): def getNumberofObjects(client, currentDay):
objects = client.list_objects(bucket_name, prefix=currentDay+'/') objects = client.list_objects(config.bucket_name, prefix=currentDay+'/')
return sum(1 for _ in objects) return sum(1 for _ in objects)
def getImageName(currentDay, client): def getObjectExtension(client, currentDay, fileNumber):
maxFiles = getNumberofObjects(client, currentDay, config.bucket_name) 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 getFileNames(currentDay, client, username):
maxFiles = getNumberofObjects(client, currentDay)
fileNumber = randint(1, maxFiles) fileNumber = randint(1, maxFiles)
desiredFile = currentDay + '/' + str(fileNumber) + '.jpeg' fileExtension = getObjectExtension(client, currentDay, fileNumber)
return desiredFile desiredFile = currentDay + '/' + str(fileNumber) + '.' + fileExtension
downloadName = username + '.' + fileExtension
return desiredFile, downloadName
def downloadImage(currentDay, username): def downloadImage(currentDay, username):
client = _setClient() client = _setClient()
client.fget_object(config.bucket_name, getImageName(currentDay, client), username + '.jpeg') object_name, file_name = getFileNames(currentDay, client, username)
client.fget_object(config.bucket_name, object_name, file_name)
def downloadForAll(currentDay): def downloadForAll(currentDay):
cur, conn = DBwork.set_connection() cur, conn = DBwork.set_connection()