← Back to branch summary

~ubuntu-branches/ubuntu/quantal/openerp6.1/quantal-proposed

« back to all changes in this revision

Viewing changes to openerp/addons/base_module_record/wizard/base_module_save.py

  • Committer: Package Import Robot
  • Author(s): Yolanda Robla
  • Date: 2012-09-20 15:29:00 UTC
  • Revision ID: package-import@ubuntu.com-20120920152900-woyy3yww8z6acmsk
Tags: upstream-6.1-1+dfsg
Import upstream version 6.1-1+dfsg

Show diffs side-by-side

added added

removed removed

Lines of Context:
openerp/addons/base_module_record/wizard/base_module_save.py
 
1
# -*- coding: utf-8 -*-
 
2
##############################################################################
 
3
#
 
4
#    OpenERP, Open Source Management Solution
 
5
#    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
 
6
#
 
7
#    This program is free software: you can redistribute it and/or modify
 
8
#    it under the terms of the GNU Affero General Public License as
 
9
#    published by the Free Software Foundation, either version 3 of the
 
10
#    License, or (at your option) any later version.
 
11
#
 
12
#    This program is distributed in the hope that it will be useful,
 
13
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
14
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
15
#    GNU Affero General Public License for more details.
 
16
#
 
17
#    You should have received a copy of the GNU Affero General Public License
 
18
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
19
#
 
20
##############################################################################
 
21
 
 
22
import zipfile
 
23
import StringIO
 
24
import base64
 
25
 
 
26
import tools
 
27
from tools.translate import _
 
28
from osv import osv, fields
 
29
 
 
30
 
 
31
def _create_yaml(self, cr, uid, data, context=None):
 
32
    mod = self.pool.get('ir.module.record')
 
33
    try:
 
34
        res_xml = mod.generate_yaml(cr, uid)
 
35
    except Exception, e:
 
36
        raise osv.except_osv(_('Error'),_(str(e)))
 
37
    return {
 
38
    'yaml_file': base64.encodestring(res_xml),
 
39
}
 
40
    
 
41
def _create_module(self, cr, uid, ids, context=None):
 
42
    mod = self.pool.get('ir.module.record')
 
43
    res_xml = mod.generate_xml(cr, uid)
 
44
    data = self.read(cr, uid, ids, [], context=context)[0]
 
45
    s = StringIO.StringIO()
 
46
    zip = zipfile.ZipFile(s, 'w')
 
47
    dname = data['directory_name']
 
48
    data['update_name'] = ''
 
49
    data['demo_name'] = ''
 
50
    if ['data_kind'] =='demo':
 
51
        data['demo_name'] = '"%(directory_name)s_data.xml"' % data
 
52
    else:
 
53
        data['update_name'] = '"%(directory_name)s_data.xml"' % data
 
54
    data['depends'] = ','.join(map(lambda x: '"'+x+'"', mod.depends.keys()))
 
55
    _terp = """{
 
56
        "name" : "%(name)s",
 
57
        "version" : "%(version)s",
 
58
        "author" : "%(author)s",
 
59
        "website" : "%(website)s",
 
60
        "category" : "%(category)s",
 
61
        "description": \"\"\"%(description)s\"\"\",
 
62
        "depends" : [%(depends)s],
 
63
        "init_xml" : [ ],
 
64
        "demo_xml" : [ %(demo_name)s],
 
65
        "update_xml" : [%(update_name)s],
 
66
        "installable": True
 
67
} """ % data
 
68
    filewrite = {
 
69
        '__init__.py':'#\n# Generated by the OpenERP module recorder !\n#\n',
 
70
        '__openerp__.py':_terp,
 
71
        dname+'_data.xml': res_xml
 
72
    }
 
73
    for name,datastr in filewrite.items():
 
74
        info = zipfile.ZipInfo(dname+'/'+name)
 
75
        info.compress_type = zipfile.ZIP_DEFLATED
 
76
        info.external_attr = 2175008768
 
77
        if not datastr:
 
78
            datastr = ''
 
79
        zip.writestr(info, datastr)
 
80
    zip.close()
 
81
    return {
 
82
        'module_file': base64.encodestring(s.getvalue()),
 
83
        'module_filename': data['directory_name']+'-'+data['version']+'.zip'
 
84
    }
 
85
 
 
86
class base_module_save(osv.osv_memory):
 
87
    _name = 'base.module.save'
 
88
    _description = "Base Module Save"
 
89
 
 
90
    def default_get(self, cr, uid, fields, context=None):
 
91
        mod = self.pool.get('ir.module.record')
 
92
        result = {}
 
93
        info = "Details of "+str(len(mod.recording_data))+" Operation(s):\n\n"
 
94
        res = super(base_module_save, self).default_get(cr, uid, fields, context=context)
 
95
        for line in mod.recording_data:
 
96
            result.setdefault(line[0],{})
 
97
            result[line[0]].setdefault(line[1][3], {})
 
98
            result[line[0]][line[1][3]].setdefault(line[1][3], 0)
 
99
            result[line[0]][line[1][3]][line[1][3]]+=1
 
100
        for key1,val1 in result.items():
 
101
            info+=key1+"\n"
 
102
            for key2,val2 in val1.items():
 
103
                info+="\t"+key2+"\n"
 
104
                for key3,val3 in val2.items():
 
105
                    info+="\t\t"+key3+" : "+str(val3)+"\n"
 
106
        if 'info_text' in fields:
 
107
            res.update({'info_text': info})
 
108
        if 'info_status' in fields:
 
109
            info_status = mod.recording and 'record' or 'no'
 
110
            res.update({'info_status': info_status})
 
111
        return res
 
112
    
 
113
    _columns = {
 
114
        'info_text': fields.text('Information', readonly=True),
 
115
        'info_status': fields.selection([('no', 'Not Recording'),('record', 'Recording')], 'Status', readonly=True),
 
116
        'info_yaml': fields.boolean('YAML'),
 
117
    }
 
118
 
 
119
    def record_save(self, cr, uid, ids, context=None):
 
120
        data = self.read(cr, uid, ids, [], context=context)[0]
 
121
        mod = self.pool.get('ir.module.record')
 
122
        mod_obj = self.pool.get('ir.model.data')
 
123
        if len(mod.recording_data):
 
124
            if data['info_yaml']:
 
125
                mod = self.pool.get('ir.module.record')
 
126
                res=_create_yaml(self, cr, uid, data, context)
 
127
                model_data_ids = mod_obj.search(cr, uid,[('model', '=', 'ir.ui.view'), ('name', '=', 'yml_save_form_view')], context=context)
 
128
                resource_id = mod_obj.read(cr, uid, model_data_ids, fields=['res_id'], context=context)[0]['res_id']
 
129
                return {
 
130
                    'name': _('Message'),
 
131
                    'context':  {
 
132
                        'default_yaml_file': tools.ustr(res['yaml_file']),
 
133
                        },
 
134
                    'view_type': 'form',
 
135
                    'view_mode': 'form',
 
136
                    'res_model': 'base.module.record.objects',
 
137
                    'views': [(resource_id, 'form')],
 
138
                    'type': 'ir.actions.act_window',
 
139
                    'target': 'new',
 
140
                }
 
141
            else:
 
142
                model_data_ids = mod_obj.search(cr, uid,[('model', '=', 'ir.ui.view'), ('name', '=', 'info_start_form_view')], context=context)
 
143
                resource_id = mod_obj.read(cr, uid, model_data_ids, fields=['res_id'], context=context)[0]['res_id']
 
144
                return {
 
145
                    'name': _('Message'),
 
146
                    'context': context,
 
147
                    'view_type': 'form',
 
148
                    'view_mode': 'form',
 
149
                    'res_model': 'base.module.record.objects',
 
150
                    'views': [(resource_id, 'form')],
 
151
                    'type': 'ir.actions.act_window',
 
152
                    'target': 'new',
 
153
                }
 
154
        model_data_ids = mod_obj.search(cr, uid,[('model', '=', 'ir.ui.view'), ('name', '=', 'module_recording_message_view')], context=context)
 
155
        resource_id = mod_obj.read(cr, uid, model_data_ids, fields=['res_id'], context=context)[0]['res_id']
 
156
        
 
157
        return {
 
158
            'name': _('Message'),
 
159
            'context': context,
 
160
            'view_type': 'form',
 
161
            'view_mode': 'form',
 
162
            'res_model': 'base.module.record.objects',
 
163
            'views': [(resource_id, 'form')],
 
164
            'type': 'ir.actions.act_window',
 
165
            'target': 'new',
 
166
        }      
 
167
        
 
168
base_module_save()
 
169
 
 
170
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 
 
b'\\ No newline at end of file'