Table-detection / minio_services.py
Sharathhebbar24's picture
First HF_SPace APp
d28ba37
raw
history blame
1.25 kB
from pathlib import Path
from minio import Minio
from minio.error import (ResponseError, BucketAlreadyOwnedByYou,
BucketAlreadyExists)
class MINIO():
def __init__(self, HOST, ACCESS_KEY, MINIO_KEY, BUCKET_NAME, UID, op):
self.minioClient = Minio(HOST,
access_key=ACCESS_KEY,
secret_key=MINIO_KEY,
secure=False)
self.BUCKET_NAME = BUCKET_NAME
self.UID = UID
self.op = op
def upload_to_minio(self):
try:
self.minioClient.make_bucket(self.BUCKET_NAME, location="us-east-1")
except BucketAlreadyOwnedByYou as err:
pass
except BucketAlreadyExists as err:
pass
except ResponseError as err:
raise
# Put an object 'A' with contents from 'B'.
try:
self.minioClient.fput_object(self.BUCKET_NAME, str(self.UID) + '/' + Path(self.op).name, self.op)
except ResponseError as err:
print(err)
def download_from_minio(self):
val = self.minioClient.fget_object(self.BUCKET_NAME, Path(self.op).name, str(self.UID) + '/' + Path(self.op).name)
return val.object_name