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
|
###############################################################################
#
# Copyright (C) 2007-TODAY Tiny ERP Pvt Ltd. All Rights Reserved.
#
# $Id$
#
# Developed by Tiny (http://openerp.com) and Axelor (http://axelor.com).
#
# The OpenERP web client is distributed under the "OpenERP Public License".
# It's based on Mozilla Public License Version (MPL) 1.1 with following
# restrictions:
#
# - All names, links and logos of Tiny, Open ERP and Axelor must be
# kept as in original distribution without any changes in all software
# screens, especially in start-up page and the software header, even if
# the application source code has been changed or updated or code has been
# added.
#
# - All distributions of the software must keep source code with OEPL.
#
# - All integrations to any other software must keep source code with OEPL.
#
# If you need commercial licence to remove this kind of restriction please
# contact us.
#
# You can see the MPL licence at: http://www.mozilla.org/MPL/MPL-1.1.html
#
###############################################################################
import re
import time
import base64
import xmlrpclib
import cherrypy
import pkg_resources
from openerp.tools import expose
from openerp.tools import validate, error_handler
from openerp.tools import redirect
from openerp import rpc
from openerp import common
from openerp import widgets
from openerp import validators
def get_lang_list():
langs = [('en_US', 'English (US)')]
try:
return langs + (rpc.session.execute_db('list_lang') or [])
except Exception, e:
pass
return langs
def get_db_list():
try:
return rpc.session.execute_db('list') or []
except:
return []
class DBForm(widgets.Form):
strip_name = True
def post_init(self, *args, **kw):
if self.validator is validators.DefaultValidator:
self.validator = validators.Schema()
for f in self.fields:
self.validator.add_field(f.name, f.validator)
class FormCreate(DBForm):
name = "create"
string = _('Create new database')
action = '/database/do_create'
submit_text = _('OK')
strip_name = True
form_attrs = {'onsubmit': 'return on_create()'}
fields = [widgets.PasswordField(name='password', label=_('Super admin password:'), validator=validators.NotEmpty()),
widgets.TextField(name='dbname', label=_('New database name:'), validator=validators.NotEmpty()),
widgets.CheckBox(name='demo_data', label=_('Load Demonstration data:'), default=True, validator=validators.Bool(if_empty=False)),
widgets.SelectField(name='language', options=get_lang_list, validator=validators.String(), label=_('Default Language:')),
widgets.PasswordField(name='admin_password', label=_('Administrator password:'), validator=validators.NotEmpty()),
widgets.PasswordField(name='confirm_password', label=_('Confirm password:'), validator=validators.NotEmpty())
]
validator = validators.Schema(chained_validators=[validators.FieldsMatch("admin_password","confirm_password")])
class FormDrop(DBForm):
name = "drop"
string = _('Drop database')
action = '/database/do_drop'
submit_text = _('OK')
form_attrs = {'onsubmit': 'return window.confirm("%s")' % _("Do you really want to drop the selected database?")}
fields = [widgets.SelectField(name='dbname', options=get_db_list, label=_('Database:'), validator=validators.String(not_empty=True)),
widgets.PasswordField(name='password', label=_('Password:'), validator=validators.NotEmpty())]
class FormBackup(DBForm):
name = "backup"
string = _('Backup database')
action = '/database/do_backup'
submit_text = _('OK')
fields = [widgets.SelectField(name='dbname', options=get_db_list, label=_('Database:'), validator=validators.String(not_empty=True)),
widgets.PasswordField(name='password', label=_('Password:'), validator=validators.NotEmpty())]
class FormRestore(DBForm):
name = "restore"
string = _('Restore database')
action = '/database/do_restore'
submit_text = _('OK')
fields = [widgets.FileField(name="filename", label=_('File:')),
widgets.PasswordField(name='password', label=_('Password:'), validator=validators.NotEmpty()),
widgets.TextField(name='dbname', label=_('New database name:'), validator=validators.NotEmpty())]
class FormPassword(DBForm):
name = "password"
string = _('Change Administrator Password')
action = '/database/do_password'
submit_text = _('OK')
fields = [widgets.PasswordField(name='old_password', label=_('Old Password:'), validator=validators.NotEmpty()),
widgets.PasswordField(name='new_password', label=_('New Password:'), validator=validators.NotEmpty()),
widgets.PasswordField(name='confirm_password', label=_('Confirm Password:'), validator=validators.NotEmpty())]
validator = validators.Schema(chained_validators=[validators.FieldsMatch("new_password","confirm_password")])
_FORMS = {
'create': FormCreate(),
'drop': FormDrop(),
'backup': FormBackup(),
'restore': FormRestore(),
'password': FormPassword()
}
class Database(object):
@expose()
def index(self, *args, **kw):
raise redirect('/database/create')
@expose(template="templates/database.mako")
def create(self, tg_errors=None, **kw):
form = _FORMS['create']
return dict(form=form)
@expose()
@validate(form=_FORMS['create'])
@error_handler(create)
def do_create(self, password, dbname, admin_password, confirm_password, demo_data=False, language=None, **kw):
if not re.match('^[a-zA-Z][a-zA-Z0-9_]+$', dbname):
raise common.warning(_('The database name must contain only normal characters or "_".\nYou must avoid all accents, space or special characters.'), _('Bad database name!'))
ok = False
try:
res = rpc.session.execute_db('create', password, dbname, demo_data, language, admin_password)
while True:
try:
progress, users = rpc.session.execute_db('get_progress', password, res)
if progress == 1.0:
for x in users:
if x['login'] == 'admin':
rpc.session.login(dbname, 'admin', password)
ok = True
break
else:
time.sleep(1)
except:
raise Exception('DbFailed')
except Exception, e:
if e.args == ('DbExist',):
raise common.warning(_("Could not create database."), _('Database already exists!'))
elif e.args == ('DbFailed'):
raise common.warning(_("The server crashed during installation.\nWe suggest you to drop this database."),
_("Error during database creation!"))
elif getattr(e, 'faultCode', False) == 'AccessDenied':
raise common.warning(_('Bad database administrator password!'), _("Could not create database."))
else:
raise common.warning(_("Could not create database."))
if ok:
raise redirect('/')
raise redirect('/login', db=dbname)
@expose(template="templates/database.mako")
def drop(self, tg_errors=None, **kw):
form = _FORMS['drop']
return dict(form=form)
@expose()
@validate(form=_FORMS['drop'])
@error_handler(drop)
def do_drop(self, dbname, password, **kw):
try:
rpc.session.execute_db('drop', password, dbname)
except Exception, e:
if getattr(e, 'faultCode', False) == 'AccessDenied':
raise common.warning(_('Bad database administrator password!'), _("Could not drop database."))
else:
raise common.warning(_("Couldn't drop database"))
raise redirect("/database/drop")
@expose(template="templates/database.mako")
def backup(self, tg_errors=None, **kw):
form = _FORMS['backup']
return dict(form=form)
@expose()
@validate(form=_FORMS['backup'])
@error_handler(backup)
def do_backup(self, dbname, password, **kw):
try:
res = rpc.session.execute_db('dump', password, dbname)
if res:
cherrypy.response.headers['Content-Type'] = "application/data"
cherrypy.response.headers['Content-Disposition'] = 'filename="' + dbname + '.dump"';
return base64.decodestring(res)
except Exception, e:
raise common.warning(_("Could not create backup."))
raise redirect('/login')
@expose(template="templates/database.mako")
def restore(self, tg_errors=None, **kw):
form = _FORMS['restore']
return dict(form=form)
@expose()
@validate(form=_FORMS['restore'])
@error_handler(restore)
def do_restore(self, filename, password, dbname, **kw):
try:
data = base64.encodestring(filename.file.read())
res = rpc.session.execute_db('restore', password, dbname, data)
except Exception, e:
if getattr(e, 'faultCode', False) == 'AccessDenied':
raise common.warning(_('Bad database administrator password!'), _("Could not restore database."))
else:
raise common.warning(_("Couldn't restore database"))
raise redirect('/login', db=dbname)
@expose(template="templates/database.mako")
def password(self, tg_errors=None, **kw):
form = _FORMS['password']
return dict(form=form)
@validate(form=_FORMS['password'])
@error_handler(password)
@expose()
def do_password(self, old_password, new_password, confirm_password, **kw):
try:
res = rpc.session.execute_db('change_admin_password', old_password, new_password)
except Exception,e:
if getattr(e, 'faultCode', False) == 'AccessDenied':
raise common.warning(_("Could not change super admin password."), _('Bad password provided!'))
else:
raise common.warning(_("Error, password not changed."))
raise redirect('/login')
# vim: ts=4 sts=4 sw=4 si et
|