73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# This file is part of the cashbook-module from m-ds for Tryton.
|
||
|
|
# The COPYRIGHT file at the top level of this repository contains the
|
||
|
|
# full copyright notices and license terms.
|
||
|
|
|
||
|
|
from quiffen import Qif
|
||
|
|
|
||
|
|
file_name = 'bargeld.qif'
|
||
|
|
company_name = 'm-ds'
|
||
|
|
|
||
|
|
from proteus import config, Model
|
||
|
|
from datetime import date
|
||
|
|
|
||
|
|
cfg1 = config.set_trytond(
|
||
|
|
'postgresql://postgres:test1@localhost:5432/tr44/',
|
||
|
|
user='admin',
|
||
|
|
config_file='/home/trytproj/projekt/py3tr60/etc/trytond.conf')
|
||
|
|
|
||
|
|
# check active modules
|
||
|
|
need_modules = ['cashbook', 'party', 'company']
|
||
|
|
|
||
|
|
IrModule = Model.get('ir.module')
|
||
|
|
mod_lst = IrModule.find([
|
||
|
|
('state', '=', 'activated'),
|
||
|
|
('name', 'in', need_modules),
|
||
|
|
])
|
||
|
|
if not len(mod_lst) == len(need_modules):
|
||
|
|
raise ValueError('einige module sind nicht aktiviert!')
|
||
|
|
|
||
|
|
|
||
|
|
def add_category(category, parent, company):
|
||
|
|
""" check or add category
|
||
|
|
"""
|
||
|
|
Category = Model.get('cashbook.category')
|
||
|
|
|
||
|
|
query = [('name', '=', category.name)]
|
||
|
|
if parent is None:
|
||
|
|
query.append(('parent', '=', None))
|
||
|
|
else :
|
||
|
|
query.append(('parent.id', '=', parent.id))
|
||
|
|
|
||
|
|
cat1 = Category.find(query)
|
||
|
|
if len(cat1) == 1:
|
||
|
|
cashbook_category = cat1[0]
|
||
|
|
print('- found:', cashbook_category.rec_name)
|
||
|
|
elif len(cat1) == 0:
|
||
|
|
print('- new-go:', getattr(parent, 'rec_name', '-'), category.name, category.income, category.expense)
|
||
|
|
cashbook_category, = Category.create([{
|
||
|
|
'name': category.name,
|
||
|
|
'cattype': 'in' if category.income == True else 'out',
|
||
|
|
'parent': getattr(parent, 'id', None),
|
||
|
|
}], context={'company': company.id})
|
||
|
|
cashbook_category = Category(cashbook_category)
|
||
|
|
print('- new-ok:', cashbook_category.rec_name)
|
||
|
|
else :
|
||
|
|
raise ValueError('invalid num of category')
|
||
|
|
|
||
|
|
for subcat in category.children:
|
||
|
|
add_category(subcat, cashbook_category, company)
|
||
|
|
|
||
|
|
# end add_category
|
||
|
|
|
||
|
|
|
||
|
|
qif = Qif.parse(file_name, day_first=True)
|
||
|
|
|
||
|
|
Company = Model.get('company.company')
|
||
|
|
company1, = Company.find([('rec_name', '=', company_name)])
|
||
|
|
|
||
|
|
# import categories
|
||
|
|
for catname in qif.categories.keys():
|
||
|
|
category = qif.categories[catname]
|
||
|
|
add_category(category, None, company1)
|