← Back to branch summary

~unifield-team/unifield-wm/us-671-homere

« back to all changes in this revision

Viewing changes to account_tools/period.py

  • Committer: chloups208
  • Date: 2012-11-21 11:15:15 UTC
  • mto: This revision was merged to the branch mainline in revision 1340.
  • Revision ID: chloups208@chloups208-laptop-20121121111515-myqv282h6xmgh053
utp-171 modification of fields po, po line, product, so, so line

Show diffs side-by-side

added added

removed removed

Lines of Context:
account_tools/period.py
 
1
#!/usr/bin/env python
 
2
# -*- coding: utf-8 -*-
 
3
##############################################################################
 
4
#
 
5
#    OpenERP, Open Source Management Solution
 
6
#    Copyright (C) 2011 TeMPO Consulting, MSF. All Rights Reserved
 
7
#    Developer: Olivier DOSSMANN
 
8
#
 
9
#    This program is free software: you can redistribute it and/or modify
 
10
#    it under the terms of the GNU Affero General Public License as
 
11
#    published by the Free Software Foundation, either version 3 of the
 
12
#    License, or (at your option) any later version.
 
13
#
 
14
#    This program is distributed in the hope that it will be useful,
 
15
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
16
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
17
#    GNU Affero General Public License for more details.
 
18
#
 
19
#    You should have received a copy of the GNU Affero General Public License
 
20
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
21
#
 
22
##############################################################################
 
23
 
 
24
from time import strftime
 
25
from osv import osv
 
26
 
 
27
def get_period_from_date(self, cr, uid, date=False, context=None):
 
28
    """
 
29
    Get period in which this date could go into, otherwise return last open period
 
30
    """
 
31
    # Some verifications
 
32
    if not context:
 
33
        context = {}
 
34
    if not date:
 
35
        return False
 
36
    # Search period in which this date come from
 
37
    period_ids = self.pool.get('account.period').search(cr, uid, [('date_start', '<=', date), ('date_stop', '>=', date)], limit=1, 
 
38
        order='date_start asc, name asc', context=context) or []
 
39
    # Get last period if no period found
 
40
    if not period_ids:
 
41
        period_ids = self.pool.get('account.period').search(cr, uid, [('state', '=', 'open')], limit=1, 
 
42
            order='date_stop desc, name desc', context=context) or []
 
43
    if isinstance(period_ids, (int, long)):
 
44
        period_ids = [period_ids]
 
45
    return period_ids
 
46
 
 
47
def get_date_in_period(self, cr, uid, date=None, period_id=None, context=None):
 
48
    """
 
49
    Permit to return a date included in period :
 
50
     - if given date is included in period, return the given date
 
51
     - else return the date_stop of given period
 
52
    """
 
53
    if not context:
 
54
        context = {}
 
55
    if not date or not period_id:
 
56
        return False
 
57
    period = self.pool.get('account.period').browse(cr, uid, period_id, context=context)
 
58
    if date < period.date_start or date > period.date_stop:
 
59
        return period.date_stop
 
60
    return date
 
61
 
 
62
class account_period(osv.osv):
 
63
    _name = 'account.period'
 
64
    _inherit = 'account.period'
 
65
 
 
66
    def get_period_from_date(self, cr, uid, date=False, context=None):
 
67
        return get_period_from_date(self, cr, uid, date, context)
 
68
 
 
69
    def get_date_in_period(self, cr, uid, date=None, period_id=None, context=None):
 
70
        return get_date_in_period(self, cr, uid, date, period_id, context)
 
71
 
 
72
account_period()
 
73
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: