120 lines
5.0 KiB
Python
120 lines
5.0 KiB
Python
from osv import osv, fields
|
|
import gdata.docs.data
|
|
import gdata.docs.client
|
|
from gdata.client import RequestError
|
|
|
|
class google_docs(osv.osv):
|
|
_name = 'google.docs'
|
|
|
|
_table = 'google_docs_templates'
|
|
_columns = {
|
|
'id': fields.integer('ID', readonly=True),
|
|
'model': fields.many2one('ir.model', 'Model'),
|
|
'gdocs_res_id': fields.char('Google resource ID', size=64, translate=False)
|
|
}
|
|
|
|
edit_url_template = 'https://docs.google.com/Edit?docid=%s'
|
|
prefix_gdoc_id_res = 'document:'
|
|
|
|
def copy_gdoc(self, cr, uid, model, context=None):
|
|
'''Associate a copy of the gdoc identified by 'gdocs_res_id' to the current entity.
|
|
@param cr: the current row from the database cursor.
|
|
@param uid: the current user ID, for security checks.
|
|
@param model: the current model name.
|
|
@param context: a standard dictionary for contextual values.
|
|
@return the url of the copy itself.
|
|
@return -1 if the template hasn't been assigned yet.
|
|
@return -2 if the google_base_account hasn't been configured yet.'''
|
|
|
|
if context==None:
|
|
context={}
|
|
|
|
# check template for the current model
|
|
model_obj = self.pool.get(model)
|
|
res_gdocs_obj = self.pool.get('google.docs')
|
|
domain = [('model' , '=', model_obj)]
|
|
gdoc = res_gdocs_obj.search(cr,uid,domain,context=context)
|
|
if not gdoc:
|
|
return -1
|
|
|
|
# check google_base_account
|
|
users_obj = self.pool.get('res.users')
|
|
user = users_obj.browse(cr, uid, [uid])[0]
|
|
if not user.gmail_user or not user.gmail_password:
|
|
return -2
|
|
|
|
# copy the document
|
|
client = gdata.docs.client.DocsClient(source='openerp.com')
|
|
client.ssl = True
|
|
client.http_client.debug = False
|
|
client.ClientLogin(user.gmail_user, user.gmail_password, client.source, service='writely')
|
|
resource = client.get_resource_by_id(gdoc.gdocs_res_id)
|
|
copied_resource = client.copy_resource(entry=resource, title=resource.title.text + ' (generated from OpenERP)')
|
|
|
|
return self.edit_url_template % (copied_resource.resource_id.text,)
|
|
|
|
def get_documents_list(self, cr, uid, context=None):
|
|
'''Return the list of google documents available at the user's account.
|
|
@param cr: the current row from the database cursor.
|
|
@param uid: the current user ID, for security checks.
|
|
@param context: a standard dictionary for contextual values.
|
|
@return a list with information about the documents in form of tuples (document_name, document_resource_id).
|
|
@return -2 if the google_base_account hasn't been configured yet.'''
|
|
|
|
if context == None:
|
|
context = {}
|
|
|
|
# check google_base_account
|
|
users_obj = self.pool.get('res.users')
|
|
user = users_obj.browse(cr, uid, [uid])[0]
|
|
if not user.gmail_user or not user.gmail_password:
|
|
return -2
|
|
|
|
# get the documents list
|
|
client = gdata.docs.client.DocsClient(source='openerp.com')
|
|
client.ssl = True
|
|
client.http_client.debug = False
|
|
client.ClientLogin(user.gmail_user, user.gmail_password, client.source, service='writely')
|
|
|
|
return map(lambda doc: (doc.title.text, doc.resource_id.text[len(prefix_gdoc_id_res):]), filter(lambda r: r.resource_id.text.startswith('document:'), client.get_all_resources()))
|
|
|
|
|
|
def set_model_document_template(self, cr, uid, model, resource_id, context=None):
|
|
'''Set the default document template for the specified model. This template doesn't have to be a google documents template itself, but just a document.
|
|
@param cr: current row for the database cursor.
|
|
@param uid: the current user ID, for security checks.
|
|
@param model: the current model name.
|
|
@param resource_id: resource_id associated to the chosen document.
|
|
@param context: a standard dictionary for contextual values.
|
|
@return 0 on successful execution.
|
|
@return -2 if the google_base_account hasn't been configured yet.
|
|
@return -3 if the given resource_id doesn't exist in the user's google docs account.
|
|
'''
|
|
|
|
if context == None:
|
|
context = {}
|
|
|
|
# check google_base_account
|
|
users_obj = self.pool.get('res.users')
|
|
user = users_obj.browse(cr, uid, [uid])[0]
|
|
if not user.gmail_user or not user.gmail_password:
|
|
return -2
|
|
|
|
# check resource_id
|
|
client = gdata.docs.client.DocsClient(source='openerp.com')
|
|
client.ssl = True
|
|
client.http_client.debug = False
|
|
client.ClientLogin(user.gmail_user, user.gmail_password, client.source, service='writely')
|
|
try:
|
|
client.get_resource_by_id(resource_id)
|
|
except RequestError:
|
|
return -3
|
|
|
|
# set the model document template
|
|
model_template_id = self.create(cr, uid,
|
|
{ 'model': self.pool.get(model),
|
|
'gdocs_res_id': resource_id
|
|
})
|
|
|
|
return 0
|