def get(self, request, *args, **kwargs):
table_name = kwargs.get('table_name', None)
list_filter = kwargs.get('list_filter', None)
list_column = kwargs.get('list_column', None)
connection_id = kwargs.get('connection', None)
try:
provider_connection = self.queryset.get(id=connection_id)
provider = provider_connection.provider
if provider:
if provider.name == MONGO:
mongo_db_manager = MongoDBManager()
try:
db, cache_db = mongo_db_manager.connection_mongo_by_provider(
provider_connection=provider_connection)
# columns = mongo_db_manager.get_all_keys(db=db, collection=table_name)
# documents, count = mongo_db_manager.get_all_documents(db=db, collection=table_name,
# column_sort=None,
# sort=None, page=1, page_size=20)
documents = mongo_db_manager.export_db_by_column(db=db, table=table_name,
list_filter=list_filter,
list_column=list_column)
data = list(documents)
result = json.loads(dumps(data))
# final_data = []
# for d in result:
# i = []
# for k,v in d.items():
# i.append(v)
# final_data.append(i)
pdf = GeneratePdf(result, table_name, list_column)
response = pdf.generate_pdf(context={})
return response
except Exception as err:
return responses.bad_request(data=str(err), message_code='BD_ERROR')
else:
# TODO: implement another phase
pass
else:
return responses.bad_request(data='Provider not found', message_code='PROVIDER_NOT_FOUND')
except DBProviderConnection.DoesNotExist as err:
return responses.not_found(data=None, message_code='EXPORT_ERROR', message_system=err)
def get(self, request, *args, **kwargs):
try:
user = request.user
connection_id = kwargs.get("connection")
table_name = kwargs.get("table_name")
list_filter = kwargs.get('list_filter', None)
list_column = kwargs.get('list_column', None)
provider_connection = self.queryset.get(id=connection_id)
provider = provider_connection.provider
if provider.name == MONGO:
mongo_db_manager = MongoDBManager()
db, cache_db = mongo_db_manager.connection_mongo_by_provider(provider_connection=provider_connection)
documents = mongo_db_manager.export_db_by_column(db=db, table=table_name,
list_filter=list_filter,
list_column=list_column)
if documents.count() <= 1000:
result = json.loads(dumps(list(documents)))
headers = list(result[0].keys())
output = io.BytesIO()
workbook = xlsxwriter.Workbook(output)
worksheet = workbook.add_worksheet()
cell_format_header = workbook.add_format()
cell_format_header.set_bold()
for index in range(len(headers)):
worksheet.write(0, index, headers[index], cell_format_header)
for row_num, columns in enumerate(result):
for index in range(len(headers)):
value = columns.get(headers[index]) if index != 0 else columns.get(headers[index]).get(
'$oid')
worksheet.write(row_num + 1, index, value)
workbook.close()
output.seek(0)
today = datetime.now().strftime("%d%m%Y_%H%M%S")
filename = f"ExportData-{table_name}-{today}.xlsx"
response = HttpResponse(
output,
content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
)
response['Content-Disposition'] = 'attachment; filename=%s' % filename
return response
else:
ExportData.objects.create(
provider_connection_id=provider_connection.id,
username=user.username,
table=table_name,
status=ExportData.INIT,
file_type=ExportData.EXCEL,
list_filter=list_filter,
list_column=list_column
)
return responses.ok(data="Waiting notify for export", method=constant.GET,
entity_name='export-data')
return responses.bad_request(data=None, message_code="SQL_PROVIDER_NOT_FOUND")
except Exception as err:
return responses.not_found(data=None, message_code='SQL_FUNCTION_NOT_FOUND', message_system=err)
def get(self, request, *args, **kwargs):
try:
user = request.user
connection_id = kwargs.get("connection")
table_name = kwargs.get("table_name")
list_filter = kwargs.get('list_filter', None)
list_column = kwargs.get('list_column', None)
provider_connection = self.queryset.get(id=connection_id)
provider = provider_connection.provider
if provider.name == MONGO:
mongo_db_manager = MongoDBManager()
db, cache_db = mongo_db_manager.connection_mongo_by_provider(provider_connection=provider_connection)
# c = db.__getattr__(table_name).find().limit(20)
documents = mongo_db_manager.export_db_by_column(db=db, table=table_name,
list_filter=list_filter,
list_column=list_column)
if documents.count() <= 1000:
result = json.loads(dumps(list(documents)))
headers = list(result[0].keys())
content = ''
for header in headers:
content += header
if headers.index(header) != len(headers) - 1:
content += ', '
else:
content += '\n'
for value in result:
for header in headers:
if header == "_id":
content += value.get(header).get('$oid')
else:
try:
content += value.get(header)
except:
content += ''
if headers.index(header) != len(headers) - 1:
content += ', '
else:
content += '\n'
today = datetime.now().strftime("%d%m%Y_%H%M%S")
filename = f"ExportData-{table_name}-{today}.txt"
response = HttpResponse(content, content_type='text/plain')
response['Content-Disposition'] = 'attachment; filename=%s' % filename
return response
else:
ExportData.objects.create(
provider_connection_id=provider_connection.id,
username=user.username,
table=table_name,
status=ExportData.INIT,
file_type=ExportData.EXCEL,
list_filter=list_filter,
list_column=list_column
)
return responses.ok(data="Waiting notify for export", method=constant.GET,
entity_name='export-data')
return responses.bad_request(data=None, message_code="SQL_PROVIDER_NOT_FOUND")
except Exception as err:
return responses.not_found(data=None, message_code='SQL_FUNCTION_NOT_FOUND', message_system=err)