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.
128 lines
5.2 KiB
128 lines
5.2 KiB
# -*- coding: utf-8 -*-
|
|
#############################################################################
|
|
#
|
|
# Cybrosys Technologies Pvt. Ltd.
|
|
#
|
|
# Copyright (C) 2024-TODAY Cybrosys Technologies(<https://www.cybrosys.com>)
|
|
# Author: Cybrosys Techno Solutions(<https://www.cybrosys.com>)
|
|
#
|
|
# You can modify it under the terms of the GNU LESSER
|
|
# GENERAL PUBLIC LICENSE (LGPL 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 LESSER GENERAL PUBLIC LICENSE (LGPL v3) for more details.
|
|
#
|
|
# You should have received a copy of the GNU LESSER GENERAL PUBLIC LICENSE
|
|
# (LGPL v3) along with this program.
|
|
# If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
#############################################################################
|
|
import base64
|
|
import binascii
|
|
import csv
|
|
import io
|
|
import tempfile
|
|
import xlrd
|
|
from datetime import date
|
|
from odoo import fields, models, _
|
|
from odoo.exceptions import ValidationError
|
|
|
|
|
|
class ImportTask(models.TransientModel):
|
|
"""For handling importing of tasks"""
|
|
_name = 'import.task'
|
|
_description = 'Importing of task'
|
|
|
|
name = fields.Char(string="Name", help="Name", default="Import Tasks")
|
|
file_type = fields.Selection([('csv', 'CSV File'),
|
|
('xls', 'XLS File')],
|
|
string='Select File Type', default='csv',
|
|
help='File type')
|
|
file_upload = fields.Binary(string="Upload File", help="Upload your file")
|
|
user_id = fields.Many2one('res.users', string='Assigned to',
|
|
help="assigned to user")
|
|
|
|
def make_json_dict(self, column, row):
|
|
""""Converting json data to dictionary"""
|
|
return [{col: item[i] for i, col in enumerate(column)} for item in row]
|
|
|
|
def action_task_import(self):
|
|
"""Creating task record using uploaded xl/csv files"""
|
|
res_partner = self.env['res.partner']
|
|
project_project = self.env['project.project']
|
|
project_task = self.env['project.task']
|
|
datas = {}
|
|
if self.file_type == 'csv':
|
|
try:
|
|
csv_data = base64.b64decode(self.file_upload)
|
|
data_file = io.StringIO(csv_data.decode("utf-8"))
|
|
data_file.seek(0)
|
|
datas = csv.DictReader(data_file, delimiter=',')
|
|
except:
|
|
raise ValidationError(_(
|
|
"File not Valid.\n\nPlease check the type and format "
|
|
"of the file and try again!"))
|
|
if self.file_type == 'xls':
|
|
try:
|
|
book_dict = {}
|
|
fp = tempfile.NamedTemporaryFile(delete=False,
|
|
suffix=".xlsx")
|
|
fp.write(binascii.a2b_base64(self.file_upload))
|
|
fp.seek(0)
|
|
book = xlrd.open_workbook(fp.name)
|
|
sheets = book.sheets()
|
|
for sheet in sheets:
|
|
book_dict[sheet.name] = {}
|
|
columns = sheet.row_values(0)
|
|
rows = []
|
|
for row_index in range(1, sheet.nrows):
|
|
row = sheet.row_values(row_index)
|
|
rows.append(row)
|
|
sheet_data = self.make_json_dict(columns, rows)
|
|
book_dict[sheet.name] = sheet_data
|
|
datas = book_dict['Sheet1']
|
|
except:
|
|
raise ValidationError(_(
|
|
"File not Valid.\n\nPlease check the type and format "
|
|
"of the file and try again!"))
|
|
for item in datas:
|
|
vals = {}
|
|
if item.get('Project'):
|
|
project = project_project.search(
|
|
[('name', '=', item.get('Project'))])
|
|
if not project:
|
|
project = project_project.create({
|
|
'name': item.get('Project')
|
|
})
|
|
vals['project_id'] = project.id
|
|
if item.get('Title'):
|
|
vals['name'] = item.get('Title')
|
|
if item.get('Customer'):
|
|
partner = res_partner.search(
|
|
[['name', '=', item.get('Customer')]])
|
|
if not partner:
|
|
partner = res_partner.create({
|
|
'name': item.get('Customer')
|
|
})
|
|
vals['partner_id'] = partner.id
|
|
if item.get('Deadline'):
|
|
if self.file_type == 'xlsx':
|
|
vals['date_deadline'] = date.fromordinal(
|
|
date(1900, 1, 1).toordinal() + int(
|
|
item.get('Deadline')) - 2)
|
|
else:
|
|
vals['date_deadline'] = item.get('Deadline')
|
|
if item.get('Parent Task'):
|
|
vals['parent_id'] = project_task.search(
|
|
[('name', '=', item.get('Parent Task'))])
|
|
vals['user_ids'] = self.user_id
|
|
project_task.create(vals)
|
|
return {
|
|
'effect': {
|
|
'fadeout': 'slow',
|
|
'message': 'Imported Successfully',
|
|
'type': 'rainbow_man',
|
|
}
|
|
}
|
|
|