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.
90 lines
3.9 KiB
90 lines
3.9 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 requests
|
|
from odoo import exceptions, fields, models, _
|
|
|
|
|
|
class UploadFile(models.TransientModel):
|
|
""" Model for opening the file upload wizard view."""
|
|
|
|
_name = "upload.file"
|
|
_description = "Upload File"
|
|
|
|
file = fields.Binary(string="Attachment", help="Select a file to upload")
|
|
file_name = fields.Char(string="File Name",
|
|
help="Name of the file to upload")
|
|
|
|
def action_upload_file(self):
|
|
"""Upload file to OneDrive."""
|
|
if not self.file:
|
|
raise exceptions.UserError(_('Please attach a file to upload.'))
|
|
# Searching for attachments related to the current upload.file record
|
|
attachment = self.env["ir.attachment"].search(
|
|
['|', ('res_field', '!=', False), ('res_field', '=', False),
|
|
('res_id', '=', self.id),
|
|
('res_model', '=', 'upload.file')])
|
|
# Getting the access token and folder ID from the onedrive.dashboard
|
|
# and ir.config_parameter models
|
|
token = self.env['onedrive.dashboard'].search([], order='id desc',
|
|
limit=1)
|
|
folder = self.env['ir.config_parameter'].get_param(
|
|
'onedrive_integration_odoo.folder_id', '')
|
|
if not token or not folder:
|
|
raise exceptions.UserError(
|
|
_('Please set up Access Token and Folder ID.'))
|
|
# Refreshing the token if it has expired
|
|
if token.token_expiry_date <= str(fields.Datetime.now()):
|
|
token.generate_onedrive_refresh_token()
|
|
try:
|
|
# Creating an upload session for the file
|
|
url = "http://graph.microsoft.com/v1.0/me/drive/items/%s:/%s:/" \
|
|
"createUploadSession" % (folder, self.file_name)
|
|
upload_session = requests.post(url, headers={
|
|
'Content-Type': 'application/json',
|
|
'Authorization': 'Bearer "' + token.onedrive_access_token + '"'
|
|
})
|
|
upload_url = upload_session.json().get('uploadUrl')
|
|
# Uploading the file to the upload session URL
|
|
requests.put(upload_url, data=open(
|
|
(attachment._full_path(attachment.store_fname)), 'rb'))
|
|
# Returning a success notification
|
|
return {
|
|
'type': 'ir.actions.client',
|
|
'tag': 'display_notification',
|
|
'params': {
|
|
'type': 'success',
|
|
'message': _(
|
|
'File has been uploaded successfully. Please refresh.'),
|
|
'next': {'type': 'ir.actions.act_window_close'},
|
|
}
|
|
}
|
|
except Exception as error:
|
|
# Returning a warning notification if the upload fails
|
|
return {
|
|
'type': 'ir.actions.client',
|
|
'tag': 'display_notification',
|
|
'params': {
|
|
'type': 'warning',
|
|
'message': _('Failed to upload: %s' % error),
|
|
'next': {'type': 'ir.actions.act_window_close'},
|
|
}
|
|
}
|
|
|