##############################################################################
#
# Copyright (c) 2004 TINY SPRL. (http://tiny.be) All Rights Reserved.
#
# $Id: subscription.py 1005 2005-07-25 08:41:42Z nicoe $
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################

#
# SPEC: Execute 'model.function(*pickle.loads(args)) periodcaly
#	date        : date to execute the job or NULL if directly
#	delete_after: delete the ir.cron entry after execution
#	interval_*  : period
#	max_repeat  : number of execution or NULL if endlessly
#
# TODO:
#	Error treatment: exception, request, ... -> send request to user_id
#

from mx import DateTime
import time
import netsvc
import tools
import sql_db
from osv import fields,osv

class subscription_document(osv.osv):
	_name = "subscription.document"
	_description = "Subscription document"
	_columns = {
		'name': fields.char('Name', size=60, required=True),
		'active': fields.boolean('Active'),
		'model': fields.char('Model', size=40),
		'workflow_action': fields.char('Workflow Action', size=40),
		'field_ids': fields.one2many('subscription.document.fields', 'document_id', 'Fields')
	}
	_defaults = {
		'active' : lambda *a: True,
	}
subscription_document()

class subscription_document_line(osv.osv):
	_name = "subscription.document.fields"
	_description = "Subscription document fields"
	_columns = {
		'name': fields.char('Name', size=60, required=True),
		'field': fields.char('Field name', size=60),
		'value': fields.selection([('false','/'),('date','Current Date')], 'Default Value', size=40),
		'document_id': fields.many2one('subscription.document', 'Subscription Document', ondelete='cascade'),
	}
	_defaults = {}
subscription_document_line()

class subscription_subscription(osv.osv):
	_name = "subscription.subscription"
	_description = "Subscription"
	_columns = {
		'name': fields.char('Name', size=60, required=True),
		'active': fields.boolean('Active'),
		'partner_id': fields.many2one('res.partner', 'Partner'),
		'notes': fields.text('Notes'),
		'user_id': fields.many2one('res.users', 'User', required=True),
		'interval_number' : fields.integer('Interval Qty'),
		'interval_type' : fields.selection([('days', 'Days'), ('weeks', 'Weeks'), ('months', 'Months')], 'Interval Unit'),
		'exec_init': fields.integer('Number of documents'),
		'date_init' : fields.datetime('First Date'),
		'state' : fields.selection([('draft','Draft'),('running','Running'),('done','Done')], 'State'),
		'doc_source': fields.char('Source Document', required=True, selection=[('account.invoice','Invoice'),('sale.order','Sale Order')], size=128),
		'doc_lines': fields.one2many('subscription.subscription.history', 'subscription_id', 'Documents created'),
		'cron_id': fields.many2one('ir.cron', 'Cron Job')
	}
	_defaults = {
		'date_init' : lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
		'user_id' : lambda x,y,z,c: z,
		'active' : lambda *a: True,
		'interval_number' : lambda *a: 1,
		'interval_type' : lambda *a: 'months',
		'doc_source' : lambda *a: False,
		'state' : lambda *a: 'draft'
	}

	def set_process(self, cr, uid, ids, context={}):
		for row in self.read(cr, uid, ids):
			mapping = {'name':'name','interval_number':'interval_number','interval_type':'interval_type','exec_init':'numbercall','date_init':'nextcall'}
			res = {'model':'subscription.subscription', 'args':'[['+str(row['id'])+']]', 'function':'model_copy', 'priority':6, 'user_id':row['user_id'] and row['user_id'][0]}
			for key,value in mapping.items():
				res[value] = row[key]
			id = self.pool.get('ir.cron').create(cr, uid, res)
			self.write(cr, uid, [row['id']], {'cron_id':id, 'state':'running'})
		return True

	def model_copy(self, cr, uid, ids, remaining=0):
		for row in self.read(cr, uid, ids):
			try:
				(model_name,id) = row['doc_source'].split(',')
				id = int(id)
				model = self.pool.get(model_name)
			except:
				raise osv.except_osv('Wrong Source Document !', 'Please provide another source document.\nThis one does not exist !')

			def model_copy(model_name, id, default={}):
				model = self.pool.get(model_name)
				datas = model.read(cr, uid, [id])[0]
				fields = model.fields_get(cr, uid)
				for f in fields:
					if f in default:
						datas[f]=default[f]
					elif fields[f]['type']=='function':
						del datas[f]
					elif fields[f]['type']=='many2one':
						datas[f]=datas[f] and datas[f][0]
					elif fields[f]['type']=='one2many':
						datas[f] = map(lambda id: (4,model_copy(fields[f]['relation'], id)), datas[f])
				del datas['id']
				id = self.pool.get(model_name).create(cr, uid, datas)
				return id

			default = {'state':'draft'}
			doc = self.pool.get('subscription.document')
			doc = doc.browse(cr,uid, doc.search(cr, uid, [('model','=',model_name)]))[0]
			for f in doc.field_ids:
				if f.value=='date':
					value=time.strftime('%Y-%m-%d')
				else:
					value = False
				default[f.field] = value
			del doc

			state = 'running'
			if not remaining:
				state='done'
			id=model_copy(model_name, id, default)
			self.pool.get('subscription.subscription.history').create(cr, uid, {'subscription_id': row['id'], 'name':row['name'], 'date':time.strftime('%Y-%m-%d %H:%M:%S'), 'document_id': model_name+','+str(id)})
			self.write(cr, uid, ids, {'state':state})
		return True

	def set_done(self, cr, uid, ids, context={}):
		res = self.read(cr,uid, ids, ['cron_id'])
		ids2 = [x['cron_id'][0] for x in res if x['id']]
		self.pool.get('ir.cron').write(cr, uid, ids2, {'active':False})
		self.write(cr, uid, ids, {'state':'done'})
		return True

	def set_draft(self, cr, uid, ids, context={}):
		self.write(cr, uid, ids, {'state':'draft'})
		return True
subscription_subscription()

class subscription_subscription_history(osv.osv):
	_name = "subscription.subscription.history"
	_description = "Subscription history"
	_columns = {
		'name': fields.char('Name', size=60, required=True),
		'date': fields.date('Date'),
		'subscription_id': fields.many2one('subscription.subscription', 'Subscription', ondelete='cascade'),
		'document_id': fields.char('Source Document', required=True, selection=[('account.invoice','Invoice'),('sale.order','Sale Order')], size=128),
	}
subscription_subscription_history()