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
|
# -*- encoding: utf-8 -*-
##############################################################################
#
# OpenERP, Open Source Management Solution
# Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>). All Rights Reserved
# $Id$
#
# 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 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
import pooler
import time
import netsvc
import os
import os.path
import tools
##############################################################################
# Service to record use of the database
##############################################################################
HOUR_MINI = 1.0
def check(chk_fnct):
data = {}
def check_one(db, uid, passwd):
data.setdefault(db, {})
cr = pooler.get_db(db).cursor()
try:
# Check if the database is blocked
cr.execute('SELECT name FROM use_control_db_block')
msg = cr.fetchone()
if msg:
# raise an Exception formatted for the client
# netsvc.Service.abortResponse can't be called while it's not a static method...
raise Exception('warning -- %s\n\n%s' % ('Database blocked', msg[0]))
if (uid not in data) or (data[uid] < time.time()):
data[uid] = time.time() + 3600 * HOUR_MINI
try:
cr.execute('insert into use_control_time (user_id, date, duration) values (%s,%s,%s)',
(int(uid), time.strftime('%Y-%m-%d %H:%M:%S'), HOUR_MINI))
cr.commit()
except:
pass
finally:
cr.close()
return chk_fnct(db, uid, passwd)
return check_one
# May be it's better using inheritancy and resubscribing the service
# Override the check method to store use of the database
from service import security
security.check = check(security.check)
##############################################################################
# Service to request feedback on a database usage
##############################################################################
class use_control_service(netsvc.Service):
def __init__(self, name="use_control"):
netsvc.Service.__init__(self, name)
self.joinGroup("web-services")
self.exportMethod(self.data_get)
self.exportMethod(self.block)
self.exportMethod(self.unblock)
def _get_size(self, cr, dbname):
"""Return the size of the system in Mb."""
cr.execute('select pg_database_size(%s)', (dbname,))
db_size = cr.fetchone()[0]
dir_size = 0.0
filestore = os.path.join(tools.config['root_path'], 'filestore', dbname)
if os.path.isdir(filestore):
for (path, dirs, files) in os.walk(filestore):
for file in files:
filename = os.path.join(path, file)
dir_size += os.path.getsize(filename)
return float((dir_size + db_size) / (1024*1024) + 1.0) / 1024.0
def data_get(self, password, db):
security.check_super(password)
cr = pooler.get_db(db).cursor()
cr.execute('''select
to_char(t.date, 'YYYY-MM-DD') as date,
u.name as username,
u.login as login,
sum(t.duration) as hours
from
use_control_time t
left join
res_users u on (u.id=t.user_id)
where (not uploaded) or (uploaded is null)
group by
to_char(t.date, 'YYYY-MM-DD'),
u.name,
u.login
''')
data = cr.fetchall()
cr.execute('update use_control_time t set uploaded=True where (not uploaded) or (uploaded is null)')
cr.execute('select name from ir_module_module where state=%s', ('installed',))
modules = map(lambda x: x[0], cr.fetchall())
hours = reduce(lambda x, y: x+y[3], data, 0.0)
cr.execute('select count(id) from res_users where active')
users = cr.fetchone()[0]
cr.execute('select max(date) from use_control_time')
maxdate = cr.fetchone()[0]
result = {
'details': data,
'modules': modules,
'latest_connection': maxdate or False,
'users_number': users,
'space': self._get_size(cr, db),
'hours': hours,
}
cr.commit()
cr.close()
print result
return result
def block(self, password, dbname, message):
security.check_super(password)
db, pool = pooler.get_db_and_pool(dbname)
cr = db.cursor()
try:
obj = pool.get('use.control.db.block')
obj.create(cr, 1, {'name': message})
finally:
cr.commit()
cr.close()
return True
def unblock(self, password, dbname):
security.check_super(password)
db, pool = pooler.get_db_and_pool(dbname)
cr = db.cursor()
try:
obj = pool.get('use.control.db.block')
obj.unlink(cr, 1, obj.search(cr, 1, []))
finally:
cr.commit()
cr.close()
return True
use_control_service()
|