You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
3.0 KiB
70 lines
3.0 KiB
# -*- coding: utf-8 -*-
|
|
###############################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2023-TODAY Cybrosys Technologies(<https://www.cybrosys.com>).
|
|
# Author: Aslam A K( odoo@cybrosys.com )
|
|
#
|
|
# You can modify it under the terms of the GNU AFFERO
|
|
# GENERAL PUBLIC LICENSE (AGPL v3), Version 3.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU AFFERO GENERAL PUBLIC LICENSE (AGPL v3) for more details.
|
|
#
|
|
# You should have received a copy of the GNU AFFERO GENERAL PUBLIC LICENSE
|
|
# (AGPL v3) along with this program.
|
|
# If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
###############################################################################
|
|
import boto3
|
|
import os
|
|
from odoo import models
|
|
|
|
|
|
class AmazonDashboard(models.Model):
|
|
""" Dashboard model to connect with Amazon S3 """
|
|
_name = 'amazon.dashboard'
|
|
_description = "Amazon Dashboard"
|
|
|
|
def amazon_view_files(self):
|
|
""" Fetch all files from s3 and returns it. """
|
|
access_key = self.env['ir.config_parameter'].get_param(
|
|
'amazon_s3_connector.amazon_access_key')
|
|
access_secret = self.env['ir.config_parameter'].get_param(
|
|
'amazon_s3_connector.amazon_secret_key')
|
|
bucket_name = self.env['ir.config_parameter'].get_param(
|
|
'amazon_s3_connector.amazon_bucket_name')
|
|
if not access_key or not access_secret or not bucket_name:
|
|
return False
|
|
try:
|
|
client = boto3.client('s3', aws_access_key_id=access_key,
|
|
aws_secret_access_key=access_secret)
|
|
region = client.get_bucket_location(Bucket=bucket_name)
|
|
client = boto3.client(
|
|
's3', region_name=region['LocationConstraint'],
|
|
aws_access_key_id=access_key,
|
|
aws_secret_access_key=access_secret
|
|
)
|
|
response = client.list_objects(Bucket=bucket_name)
|
|
file = []
|
|
for data in response['Contents']:
|
|
url = client.generate_presigned_url(
|
|
ClientMethod='get_object',
|
|
Params={'Bucket': bucket_name, 'Key': data['Key']})
|
|
if data['Size'] == 0:
|
|
continue
|
|
size_bytes = data['Size'] / 1024
|
|
if size_bytes > 1024:
|
|
size = str(round(data['Size'] / (1024 * 1024), 1)) + ' MB'
|
|
else:
|
|
size = str(round(data['Size'] / 1024, 1)) + ' KB'
|
|
file_type = str.upper(
|
|
os.path.splitext(data['Key'])[1].replace('.', ''))
|
|
file.append([data['Key'], url, file_type,
|
|
str(data['LastModified']), size])
|
|
return file
|
|
except Exception as error:
|
|
return ['e', error]
|
|
|