118 lines
5.2 KiB
Python
118 lines
5.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# OpenERP, Open Source Management Solution
|
|
# Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# 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 for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
##############################################################################
|
|
|
|
import time
|
|
from datetime import datetime
|
|
from dateutil.relativedelta import relativedelta
|
|
|
|
from osv import osv, fields
|
|
|
|
# Object creating periods quickly
|
|
# changed that stop_date is created with hour 23:59:00 when it was 00:00:00 stop date was excluded from period
|
|
class stock_period_createlines(osv.osv_memory):
|
|
_name = "stock.period.createlines"
|
|
|
|
def _get_new_period_start(self, cr, uid, context=None):
|
|
cr.execute("select max(date_stop) from stock_period")
|
|
result = cr.fetchone()
|
|
last_date = result and result[0] or False
|
|
if last_date:
|
|
period_start = datetime.strptime(last_date,"%Y-%m-%d %H:%M:%S")+ relativedelta(days=1)
|
|
period_start = period_start - relativedelta(hours=period_start.hour, minutes=period_start.minute, seconds=period_start.second)
|
|
else:
|
|
period_start = datetime.today()
|
|
return period_start.strftime('%Y-%m-%d')
|
|
|
|
|
|
_columns = {
|
|
'name': fields.char('Period Name', size=64),
|
|
'date_start': fields.date('Start Date', required=True, help="Starting date for planning period."),
|
|
'date_stop': fields.date('End Date', required=True, help="Ending date for planning period."),
|
|
'period_ids': fields.one2many('stock.period', 'planning_id', 'Periods'),
|
|
}
|
|
_defaults={
|
|
'date_start': _get_new_period_start,
|
|
}
|
|
|
|
def create_stock_periods(self, cr, uid, ids, context=None):
|
|
if context is None:
|
|
context = {}
|
|
interval = context.get('interval',0)
|
|
name = context.get('name','Daily')
|
|
period_obj = self.pool.get('stock.period')
|
|
lines = []
|
|
for p in self.browse(cr, uid, ids, context=context):
|
|
dt_stp = datetime.strptime(p.date_stop, '%Y-%m-%d')
|
|
ds = datetime.strptime(p.date_start, '%Y-%m-%d')
|
|
|
|
while ds <= dt_stp:
|
|
if name =='Daily':
|
|
de = ds + relativedelta(days=(interval + 1), seconds =-1)
|
|
new_id = period_obj.create(cr, uid, {
|
|
'name': de.strftime('%Y-%m-%d'),
|
|
'date_start': ds.strftime('%Y-%m-%d %H:%M:%S'),
|
|
'date_stop': de.strftime('%Y-%m-%d %H:%M:%S'),
|
|
})
|
|
ds = ds + relativedelta(days=(interval + 1))
|
|
if name =="Weekly":
|
|
de = ds + relativedelta(days=(interval + 1), seconds =-1)
|
|
if dt_stp < de:
|
|
de = dt_stp + relativedelta(days=1, seconds =-1)
|
|
else:
|
|
de = ds + relativedelta(days=(interval + 1), seconds =-1)
|
|
new_name = ds.strftime('Week %W-%Y')
|
|
if ds.strftime('%Y') != de.strftime('%Y'):
|
|
new_name = ds.strftime('Week %W-%Y') + ', ' + de.strftime('Week %W-%Y')
|
|
new_id = period_obj.create(cr, uid, {
|
|
'name': new_name,
|
|
'date_start': ds.strftime('%Y-%m-%d %H:%M:%S'),
|
|
'date_stop': de.strftime('%Y-%m-%d %H:%M:%S'),
|
|
})
|
|
ds = ds + relativedelta(days=(interval + 1))
|
|
if name == "Monthly":
|
|
de = ds + relativedelta(months=interval, seconds=-1)
|
|
if dt_stp < de:
|
|
de = dt_stp + relativedelta(days=1, seconds =-1)
|
|
else:
|
|
de = ds + relativedelta(months=interval, seconds=-1)
|
|
new_name = ds.strftime('%Y/%m')
|
|
if ds.strftime('%m') != de.strftime('%m'):
|
|
new_name = ds.strftime('%Y/%m') + '-' + de.strftime('%Y/%m')
|
|
new_id =period_obj.create(cr, uid, {
|
|
'name': new_name,
|
|
'date_start': ds.strftime('%Y-%m-%d %H:%M:%S'),
|
|
'date_stop': de.strftime('%Y-%m-%d %H:%M:%S'),
|
|
})
|
|
ds = ds + relativedelta(months=interval)
|
|
lines.append(new_id)
|
|
|
|
return {
|
|
'domain': "[('id','in', ["+','.join(map(str, lines))+"])]",
|
|
'view_type': 'form',
|
|
"view_mode": 'tree, form',
|
|
'res_model': 'stock.period',
|
|
'type': 'ir.actions.act_window',
|
|
}
|
|
|
|
stock_period_createlines()
|
|
|
|
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
|