1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
|
# -*- encoding: utf-8 -*-
##############################################################################
#
# OpenERP, Open Source Management Solution
# Copyright (C) 2004-2009 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 osv import fields
from osv import osv
import netsvc
import traceback
import sys
import datetime
from tools.translate import _
class dm_workitem(osv.osv): # {{{
_name = "dm.workitem"
_inherit = "dm.workitem"
_columns = {
'sale_order_id': fields.many2one('sale.order', 'Sale Order'),
}
def run(self, cr, uid, wi, context={}):
result = super(dm_workitem, self).run(cr, uid, wi, context)
if wi.sale_order_id:
so_res = self.pool.get('sale.order')._sale_order_process(cr, uid, wi.sale_order_id.id)
if wi.step_id.partner_event_create:
self.pool.get('res.partner.event').create(cr, uid, {
'name': wi.step_id.name,
'partner_type': 'customer',
'date': time.strftime('%Y-%m-%d %H:%M:%S'),
'partner_id':wi.address_id.partner_id.id,
'document': wi.sale_order_id and ('sale.order' +',' +'%d' %wi.sale_order_id.id) or ''})
return result
def copy(self, cr, uid, id, default=None, context=None):
if not default:
default = {}
if context is None:
context = {}
default.update({'sale_order_id': False})
copy_id = super(dm_workitem, self).copy(cr, uid, id, default, context)
return copy_id
dm_workitem() # }}}
class dm_event_sale(osv.osv_memory): # {{{
_name = "dm.event.sale"
_rec_name = "segment_id"
_columns = {
'segment_id': fields.many2one('dm.campaign.proposition.segment', 'Segment', required=True),
'step_id': fields.many2one('dm.offer.step', 'Offer Step', required=True),
'source': fields.selection([('address_id', 'Addresses')], 'Source', required=True),
'address_id': fields.many2one('res.partner.address', 'Address'),
'trigger_type_id': fields.many2one('dm.offer.step.transition.trigger', 'Trigger Condition', required=True),
'sale_order_id': fields.many2one('sale.order', 'Sale Order'),
'mail_service_id': fields.many2one('dm.mail_service', 'Mail Service'),
'action_time': fields.datetime('Action Time'),
'is_realtime': fields.boolean('Realtime Processing'),
}
_defaults = {
'source': lambda *a: 'address_id',
'sale_order_id': lambda *a: False,
}
def create(self, cr, uid, vals, context={}):
id = super(dm_event_sale, self).create(cr, uid, vals, context)
obj = self.browse(cr, uid, id)
tr_ids = self.pool.get('dm.offer.step.transition').search(cr, uid,
[('step_from_id', '=', obj.step_id.id),
('condition_id', '=', obj.trigger_type_id.id)])
if not tr_ids:
netsvc.Logger().notifyChannel('dm event case', netsvc.LOG_WARNING,
"There is no outgoing transition %s at this step: %s" % (obj.trigger_type_id.name,
obj.step_id.code))
raise osv.except_osv('Warning',
"There is no outgoing transition %s at this step: %s" % (obj.trigger_type_id.name,
obj.step_id.code))
for tr in self.pool.get('dm.offer.step.transition').browse(cr, uid, tr_ids):
if obj.action_time:
action_time = datetime.datetime.strptime(obj.action_time, '%Y-%m-%d %H:%M:%S')
else:
if obj.is_realtime:
action_time = datetime.datetime.now()
else:
wi_action_time = datetime.datetime.now()
kwargs = {(tr.delay_type+'s'): tr.delay}
action_time = wi_action_time + datetime.timedelta(**kwargs)
if tr.action_hour:
hour_str = str(tr.action_hour).split('.')[0] + \
':' + str(int(int(str(tr.action_hour).split('.')[1]) * 0.6))
act_hour = datetime.datetime.strptime(hour_str, '%H:%M')
action_time = action_time.replace(hour=act_hour.hour)
action_time = action_time.replace(minute=act_hour.minute)
if tr.action_day:
action_time = action_time.replace(day = int(tr.action_day))
if action_time.day > int(tr.action_day):
action_time = action_time.replace(month=action_time.month + 1)
if tr.action_date:
action_time = tr.action_date
try:
wi_id = self.pool.get('dm.workitem').create(cr, uid, {
'step_id': tr.step_to_id.id or False,
'segment_id': obj.segment_id.id or False,
'address_id': obj.address_id.id,
'mail_service_id': obj.mail_service_id.id,
'action_time': action_time.strftime('%Y-%m-%d %H:%M:%S'),
'tr_from_id': tr.id,
'source': obj.source,
'is_realtime': obj.is_realtime,
'sale_order_id': obj.sale_order_id.id})
except Exception, exception:
tb = sys.exc_info()
tb_s = "".join(traceback.format_exception(*tb))
netsvc.Logger().notifyChannel('dm event sale', netsvc.LOG_ERROR,
"Event cannot create Workitem: %s\n%s" % (str(exception), tb_s))
return id
dm_event_sale() # }}}
class sale_order(osv.osv): #{{{
_name = "sale.order"
_inherit = "sale.order"
_columns = {
'offer_step_id': fields.many2one('dm.offer.step', 'Offer Step', select="1"),
'segment_id': fields.many2one('dm.campaign.proposition.segment',
'Segment', select="1"),
'sale_journal_id': fields.many2one('account.journal', 'Sales Journal'),
'payment_journal_id': fields.many2one('account.journal', 'Payment Journal'),
'lines_number': fields.integer('Number of sale order lines'),
'so_confirm_do': fields.boolean('Auto confirm sale order'),
'invoice_create_do': fields.boolean('Auto create invoice'),
'invoice_validate_do': fields.boolean('Auto validate invoice'),
'invoice_pay_do': fields.boolean('Auto pay invoice'),
}
def _sale_order_process(self, cr, uid, sale_order_id):
so = self.pool.get('sale.order').browse(cr, uid, sale_order_id)
wf_service = netsvc.LocalService('workflow')
try:
if so.so_confirm_do and so.state == 'draft':
wf_service.trg_validate(uid, 'sale.order', sale_order_id,
'order_confirm', cr)
if so.invoice_create_do:
inv_id = self.pool.get('sale.order').action_invoice_create(cr, uid, [sale_order_id])
if so.sale_journal_id and so.sale_journal_id.default_credit_account_id:
self.pool.get('account.invoice').write(cr, uid, inv_id,
{'journal_id': so.sale_journal_id.id,
'account_id': so.sale_journal_id.default_credit_account_id.id})
if inv_id and so.invoice_validate_do:
wf_service.trg_validate(uid, 'account.invoice', inv_id,
'invoice_open', cr)
if so.invoice_pay_do:
ids = self.pool.get('account.period').find(cr, uid, {})
period_id = False
if len(ids):
period_id = ids[0]
cur_obj = self.pool.get('res.currency')
for invoice in so.invoice_ids:
journal = so.payment_journal_id
if not journal:
raise osv.except_osv('Error !', 'There is no payment journal defined.')
amount = invoice.residual
if journal.currency and invoice.company_id.currency_id.id != journal.currency.id:
ctx = {'date': time.strftime('%Y-%m-%d')}
amount = cur_obj.compute(cr, uid,
journal.currency.id,
invoice.company_id.currency_id.id,
amount, context=ctx)
acc_id = journal.default_credit_account_id and journal.default_credit_account_id.id
if not acc_id:
raise osv.except_osv('Error !', 'Your journal must have a default credit and debit account.')
self.pool.get('account.invoice').pay_and_reconcile(cr, uid, [invoice.id],
amount, acc_id, period_id, journal.id, '', '', '',)
except Exception, exception:
tb = sys.exc_info()
tb_s = "".join(traceback.format_exception(*tb))
wi_id = self.pool.get('dm.workitem').search(cr, uid, [('sale_order_id', '=', sale_order_id)])
self.pool.get('dm.workitem').write(cr, uid, wi_id, {'state': 'error',
'error_msg': 'Exception: %s\n%s' % (str(exception), tb_s)})
netsvc.Logger().notifyChannel('dm action - so process',
netsvc.LOG_ERROR, 'Exception: %s\n%s' % (str(exception), tb_s))
sale_order()#}}}
class dm_offer_step(osv.osv): # {{{
_name = "dm.offer.step"
_inherit = "dm.offer.step"
_columns = {
'forecasted_yield': fields.float('Forecasted Yield'),
'has_product_constraints': fields.boolean('Article Constraints', help="Tells if the choice of a product in this offer step has some constraints (weight, language, visuals, size, ...)"),
'item_ids': fields.many2many('product.product',
'dm_offer_step_product_rel', 'product_id', 'offer_step_id',
'Items', states={'closed': [('readonly', True)]},
#domain = [('product_tmpl_id.categ_id.name','=','Items' )],
context = {'dm_product_categ' : 'Direct Marketing / Items'},
),
}
dm_offer_step() # }}}
class dm_campaign_proposition(osv.osv): #{{{
_inherit = "dm.campaign.proposition"
def get_step_products(self, cr, uid, ids, *args):
prop_obj = self.browse(cr, uid, ids)[0]
offer_id = prop_obj.camp_id.offer_id.id
step_ids = self.pool.get('dm.offer.step').search(cr,
uid, [('offer_id', '=', offer_id)])
step_obj = self.pool.get('dm.offer.step').browse(cr, uid, step_ids)
if prop_obj.item_ids:
for p in prop_obj.item_ids:
self.pool.get('dm.campaign.proposition.item').unlink(cr, uid, p.id)
if not prop_obj.customer_pricelist_id:
raise osv.except_osv('Error !', 'Select a product pricelist !')
if not prop_obj.camp_id.state == 'draft':
raise osv.except_osv("Error", "Campaign Should be in draft state")
for step in step_obj:
for item in step.item_ids:
price = self.pool.get('product.pricelist').price_get(cr, uid,
[prop_obj.customer_pricelist_id.id], item.id, 1.0,)[prop_obj.customer_pricelist_id.id]
vals = {'product_id': item.id,
'proposition_id': ids[0],
'offer_step_id': step.id,
'qty_planned': item.virtual_available,
'qty_real': item.qty_available,
'price': price,
'notes': item.description,
'forecasted_yield': step.forecasted_yield,
}
self.pool.get('dm.campaign.proposition.item').create(cr, uid, vals)
return True
dm_campaign_proposition()#}}}
class product_product(osv.osv): # {{{
_inherit = "product.product"
def search(self, cr, uid, args, offset=0, limit=None, order=None, context=None, count=False):
if not context:
context = {}
if 'offer_id' in context and context['offer_id']:
result = []
offer_browse_id = self.pool.get('dm.offer').browse(cr, uid,
context['offer_id'])
for step in offer_browse_id.step_ids:
for item in step.item_ids:
result.append(item.id)
return result
if 'dm_product_categ' in context and context['dm_product_categ']:
parent_categ, categ_name = context['dm_product_categ'].split('/')
parent_id = self.pool.get('product.category').search(cr, uid,
[('name', '=', parent_categ.strip())] )
categ_id = self.pool.get('product.category').search(cr, uid,
[('name', '=', categ_name.strip()),
('parent_id', '=', parent_id)])
args.append(('categ_id','child_of',categ_id))
return super(product_product, self).search(cr, uid, args, offset, limit, order, context, count)
product_product() # }}}
class dm_campaign(osv.osv): # {{{
_inherit = "dm.campaign"
def check_forbidden_country(self, cr, uid, offer_id, country):
offers = self.pool.get('dm.offer').browse(cr, uid, offer_id)
country_name = self.pool.get('res.country').browse(cr, uid, [country])[0]
for step in offers.step_ids:
for product in step.item_ids:
for prod_country in product.country_ids:
if country == prod_country.id:
raise osv.except_osv("Error", "That product cannot be sold in %s : %s" %(country_name.name, product.code))
return True
dm_campaign() # }}}
class product_pricelist(osv.osv): # {{{
_inherit = "product.pricelist"
def search(self, cr, uid, args, offset=0, limit=None, order=None, context=None, count=False): # {{{
if not context :
context = {}
if 'camp_id' in context and context['camp_id']:
camp_obj = self.pool.get('dm.campaign').browse(cr, uid, context['camp_id'])
args.append(('currency_id','=',camp_obj.currency_id.id))
result = super(product_pricelist, self).search(cr, uid, args, offset, limit, order, context, count)
return result # }}}
product_pricelist() # }}}
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
|