298 lines
9.1 KiB
Python
298 lines
9.1 KiB
Python
# -*- coding: utf8 -*-
|
|
##############################################################################
|
|
#
|
|
# Copyright (c) 2004-2006 TINY SPRL. (http://tiny.be) All Rights Reserved.
|
|
#
|
|
# $Id: misc.py 1304 2005-09-08 14:35: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.
|
|
#
|
|
##############################################################################
|
|
|
|
"""
|
|
Miscelleanous tools used by tiny ERP.
|
|
"""
|
|
|
|
import os, time, sys
|
|
import inspect
|
|
|
|
import psycopg
|
|
import netsvc
|
|
from config import config
|
|
#import tools
|
|
|
|
if sys.version_info[:2] < (2, 4):
|
|
from threadinglocal import local
|
|
else:
|
|
from threading import local
|
|
|
|
# initialize a database with base/base.sql
|
|
def init_db(cr):
|
|
f = os.path.join(config['addons_path'], 'base/base.sql')
|
|
for line in file(f).read().split(';'):
|
|
if (len(line)>0) and (not line.isspace()):
|
|
cr.execute(line)
|
|
cr.commit()
|
|
|
|
opj = os.path.join
|
|
ad = config['addons_path']
|
|
|
|
for i in os.listdir(ad):
|
|
terp_file = opj(ad, i, '__terp__.py')
|
|
if os.path.isfile(terp_file):
|
|
info = eval(file(terp_file).read())
|
|
categs = info.get('category', 'Uncategorized').split('/')
|
|
p_id = None
|
|
while categs:
|
|
if p_id is not None:
|
|
cr.execute('select id from ir_module_category where name=%s and parent_id=%d', (categs[0], p_id))
|
|
else:
|
|
cr.execute('select id from ir_module_category where name=%s and parent_id is NULL', (categs[0],))
|
|
c_id = cr.fetchone()
|
|
if not c_id:
|
|
cr.execute('select nextval(\'ir_module_category_id_seq\')')
|
|
c_id = cr.fetchone()[0]
|
|
cr.execute('insert into ir_module_category (id, name, parent_id) values (%d, %s, %d)', (c_id, categs[0], p_id))
|
|
else:
|
|
c_id = c_id[0]
|
|
p_id = c_id
|
|
categs = categs[1:]
|
|
|
|
active = info.get('active', True)
|
|
installable = info.get('installable', True)
|
|
if installable:
|
|
if active:
|
|
state = 'to install'
|
|
else:
|
|
state = 'uninstalled'
|
|
else:
|
|
state = 'uninstallable'
|
|
cr.execute('select nextval(\'ir_module_module_id_seq\')')
|
|
id = cr.fetchone()[0]
|
|
cr.execute('insert into ir_module_module (id, author, latest_version, website, name, shortdesc, description, category_id, state) values (%d, %s, %s, %s, %s, %s, %s, %d, %s)', (
|
|
id,
|
|
info.get('author', ''),
|
|
info.get('version', ''),
|
|
info.get('website', ''),
|
|
i,
|
|
info.get('name', False),
|
|
info.get('description', ''),
|
|
p_id,
|
|
state))
|
|
dependencies = info.get('depends', [])
|
|
for d in dependencies:
|
|
cr.execute('insert into ir_module_module_dependency (module_id,name) values (%s, %s)', (id, d))
|
|
cr.commit()
|
|
|
|
def find_in_path(name):
|
|
path = [dir for dir in os.environ['PATH'].split(':')
|
|
if os.path.isdir(dir)]
|
|
for dir in path:
|
|
if name in os.listdir(dir):
|
|
return os.path.join(dir, name)
|
|
return None
|
|
|
|
def find_pg_tool(name):
|
|
if config['pg_path'] and config['pg_path'] != 'None':
|
|
return os.path.join(config['pg_path'], name)
|
|
else:
|
|
return find_in_path(name)
|
|
|
|
def exec_pg_command(name, *args):
|
|
prog = find_pg_tool(name)
|
|
args2 = (os.path.basename(prog),) + args
|
|
return os.spawnv(os.P_WAIT, prog, args2)
|
|
# os.spawnv(os.P_WAIT, prog, ([os.path.basename(prog)] + args))
|
|
|
|
def exec_pg_command_pipe(name, *args):
|
|
prog = find_pg_tool(name)
|
|
if os.name == "nt":
|
|
cmd = '"' + prog + '" ' + ' '.join(args)
|
|
else:
|
|
cmd = prog + ' ' + ' '.join(args)
|
|
return os.popen2(cmd, 'b')
|
|
|
|
#----------------------------------------------------------
|
|
# File paths
|
|
#----------------------------------------------------------
|
|
#file_path_root = os.getcwd()
|
|
#file_path_addons = os.path.join(file_path_root, 'addons')
|
|
|
|
def file_open(name, mode="r", subdir='addons'):
|
|
"""Open a file from the Tiny ERP root, using a subdir folder."""
|
|
name = os.path.join(config['root_path'], subdir, name)
|
|
return file(name, mode)
|
|
|
|
#----------------------------------------------------------
|
|
# Emails
|
|
#----------------------------------------------------------
|
|
def email_send(email_from, email_to, subject, body, email_cc=[], email_bcc=[], on_error=False, reply_to=False):
|
|
"""Send an email."""
|
|
import smtplib
|
|
from email.MIMEText import MIMEText
|
|
from email.MIMEMultipart import MIMEMultipart
|
|
from email.Header import Header
|
|
from email.Utils import formatdate, COMMASPACE
|
|
|
|
msg = MIMEText(body or '', _charset='utf-8')
|
|
msg['Subject'] = Header(subject.decode('utf8'), 'utf-8')
|
|
msg['From'] = email_from
|
|
if reply_to:
|
|
msg['Reply-To'] = msg['From']+', '+reply_to
|
|
msg['To'] = COMMASPACE.join(email_to)
|
|
if email_cc:
|
|
msg['Cc'] = COMMASPACE.join(email_cc)
|
|
if email_bcc:
|
|
msg['Bcc'] = COMMASPACE.join(email_bcc)
|
|
msg['Date'] = formatdate(localtime=True)
|
|
try:
|
|
s = smtplib.SMTP()
|
|
s.connect(config['smtp_server'])
|
|
s.sendmail(email_from, email_to + email_cc + email_bcc, msg.as_string())
|
|
s.quit()
|
|
except Exception, e:
|
|
import logging
|
|
logging.getLogger().info(str(e))
|
|
return True
|
|
|
|
#----------------------------------------------------------
|
|
# SMS
|
|
#----------------------------------------------------------
|
|
# text must be latin-1 encoded
|
|
def sms_send(user, password, api_id, text, to):
|
|
import urllib
|
|
params = urllib.urlencode({'user': user, 'password': password, 'api_id': api_id, 'text': text, 'to':to})
|
|
#print "http://api.clickatell.com/http/sendmsg", params
|
|
#f = urllib.urlopen("http://api.clickatell.com/http/sendmsg", params)
|
|
print "http://196.7.150.220/http/sendmsg", params
|
|
f = urllib.urlopen("http://196.7.150.220/http/sendmsg", params)
|
|
print f.read()
|
|
return True
|
|
|
|
#---------------------------------------------------------
|
|
# Class that stores an updateable string (used in wizards)
|
|
#---------------------------------------------------------
|
|
class UpdateableStr(local):
|
|
|
|
def __init__(self, string=''):
|
|
self.string = string
|
|
|
|
def __str__(self):
|
|
return str(self.string)
|
|
|
|
def __repr__(self):
|
|
return str(self.string)
|
|
|
|
def __nonzero__(self):
|
|
return bool(self.string)
|
|
|
|
# Don't use ! Use res.currency.round()
|
|
class currency(float):
|
|
|
|
def __init__(self, value, accuracy=2, rounding=None):
|
|
if rounding is None:
|
|
rounding=10**-accuracy
|
|
self.rounding=rounding
|
|
self.accuracy=accuracy
|
|
|
|
def __new__(cls, value, accuracy=2, rounding=None):
|
|
return float.__new__(cls, round(value, accuracy))
|
|
|
|
#def __str__(self):
|
|
# display_value = int(self*(10**(-self.accuracy))/self.rounding)*self.rounding/(10**(-self.accuracy))
|
|
# return str(display_value)
|
|
|
|
|
|
#
|
|
# Use it as a decorator of the function you plan to cache
|
|
# Timeout: 0 = no timeout, otherwise in seconds
|
|
#
|
|
class cache(object):
|
|
def __init__(self, timeout=10000, skiparg=2):
|
|
self.timeout = timeout
|
|
self.cache = {}
|
|
|
|
def __call__(self, fn):
|
|
arg_names = inspect.getargspec(fn)[0][2:]
|
|
def cached_result(self2, cr=None, *args, **kwargs):
|
|
if cr is None:
|
|
self.cache = {}
|
|
return True
|
|
|
|
# Update named arguments with positional argument values
|
|
kwargs.update(dict(zip(arg_names, args)))
|
|
kwargs = kwargs.items()
|
|
kwargs.sort()
|
|
|
|
# Work out key as a tuple of ('argname', value) pairs
|
|
key = (('dbname', cr.dbname),) + tuple(kwargs)
|
|
|
|
# Check cache and return cached value if possible
|
|
if key in self.cache:
|
|
(value, last_time) = self.cache[key]
|
|
mintime = time.time() - self.timeout
|
|
if self.timeout <= 0 or mintime <= last_time:
|
|
return value
|
|
else:
|
|
pass
|
|
#
|
|
# Clear Cache at this point ?
|
|
#
|
|
|
|
# Work out new value, cache it and return it
|
|
# Should copy() this value to avoid futur modf of the cacle ?
|
|
result = fn(self2,cr,**dict(kwargs))
|
|
|
|
self.cache[key] = (result, time.time())
|
|
return result
|
|
return cached_result
|
|
|
|
def get_languages():
|
|
languages={
|
|
'zh_CN': 'Chinese (CN)',
|
|
'zh_TW': 'Chinese (TW)',
|
|
'cs_CZ': 'Czech',
|
|
'de_DE': 'Deutsch',
|
|
'es_AR': 'Español (Argentina)',
|
|
'es_ES': 'Español (España)',
|
|
'fr_FR': 'Français',
|
|
'fr_CH': 'Français (Suisse)',
|
|
'en_EN': 'English (default)',
|
|
'hu_HU': 'Hungarian',
|
|
'it_IT': 'Italiano',
|
|
'pt_BR': 'Portugese (Brasil)',
|
|
'pt_PT': 'Portugese (Portugal)',
|
|
'nl_NL': 'Nederlands',
|
|
'ro_RO': 'Romanian',
|
|
'ru_RU': 'Russian',
|
|
'sv_SE': 'Swedish',
|
|
}
|
|
return languages
|
|
|
|
def scan_languages():
|
|
import glob
|
|
file_list = [os.path.splitext(os.path.basename(f))[0] for f in glob.glob(os.path.join(config['root_path'], 'i18n', '*.csv'))]
|
|
lang_dict = get_languages()
|
|
return [(lang, lang_dict.get(lang, lang)) for lang in file_list]
|
|
|
|
# vim:noexpandtab
|