# Copyright (c) 2012 Web Notes Technologies Pvt Ltd (http://erpnext.com)
#
# MIT License (MIT)
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the "Software"),
# to deal in the Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
# and/or sell copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
# PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
# CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
# OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#
"""
globals attached to webnotes module
+ some utility functions that should probably be moved
"""
code_fields_dict = {
'Page':[('script', 'js'), ('content', 'html'), ('style', 'css'), ('static_content', 'html'), ('server_code', 'py')],
'DocType':[('server_code_core', 'py'), ('client_script_core', 'js')],
'Search Criteria':[('report_script', 'js'), ('server_script', 'py'), ('custom_query', 'sql')],
'Patch':[('patch_code', 'py')],
'Stylesheet':['stylesheet', 'css'],
'Page Template':['template', 'html'],
'Control Panel':[('startup_code', 'js'), ('startup_css', 'css')]
}
version = 'v170'
form_dict = {}
auth_obj = None
conn = None
form = None
session = None
user = None
is_testing = None
incoming_cookies = {}
add_cookies = {} # append these to outgoing request
cookies = {}
auto_masters = {}
tenant_id = None
response = {'message':'', 'exc':''}
debug_log = []
message_log = []
class ValidationError(Exception):
pass
class AuthenticationError(Exception):
pass
class UnknownDomainError(Exception):
def __init__(self, value):
self.value = value
def __str__(self):
return repr(self.value)
class SessionStopped(Exception):
def __init__(self, value):
self.value = value
def __str__(self):
return repr(self.value)
def getTraceback():
import utils
return utils.getTraceback()
def errprint(msg):
"""
Append to the :data:`debug log`
"""
from utils import cstr
debug_log.append(cstr(msg or ''))
def msgprint(msg, small=0, raise_exception=0, as_table=False):
"""
Append to the :data:`message_log`
"""
from utils import cstr
if as_table and type(msg) in (list, tuple):
msg = '
' + ''.join([''+''.join(['%s | ' % c for c in r])+'
' for r in msg]) + '
'
message_log.append((small and '__small:' or '')+cstr(msg or ''))
if raise_exception:
raise ValidationError
def is_apache_user():
import os
if os.environ.get('USER') == 'apache':
return True
else:
return (not os.environ.get('USER'))
# os.environ does not have user, so allows a security vulnerability,fixed now.
def get_index_path():
import os
return os.sep.join(os.path.dirname(os.path.abspath(__file__)).split(os.sep)[:-2])
def get_files_path():
import defs, os
if not conn:
raise Exception, 'You must login first'
if defs.files_path:
return os.path.join(defs.files_path, conn.cur_db_name)
else:
return os.path.join(get_index_path(), 'user_files', conn.cur_db_name)
def create_folder(path):
"""
Wrapper function for os.makedirs (does not throw exception if directory exists)
"""
import os
try:
os.makedirs(path)
except OSError, e:
if e.args[0]!=17:
raise e
def create_symlink(source_path, link_path):
"""
Wrapper function for os.symlink (does not throw exception if directory exists)
"""
import os
try:
os.symlink(source_path, link_path)
except OSError, e:
if e.args[0]!=17:
raise e
def remove_file(path):
"""
Wrapper function for os.remove (does not throw exception if file/symlink does not exists)
"""
import os
try:
os.remove(path)
except OSError, e:
if e.args[0]!=2:
raise e
def connect(db_name=None):
"""
Connect to this db (or db), if called from command prompt
"""
if is_apache_user():
raise Exception, 'Not for web users!'
import webnotes.db
global conn
conn = webnotes.db.Database(user=db_name)
global session
session = {'user':'Administrator'}
import webnotes.profile
global user
user = webnotes.profile.Profile('Administrator')
def get_env_vars(env_var):
import os
return os.environ.get(env_var,'None')
remote_ip = get_env_vars('REMOTE_ADDR') #Required for login from python shell
logger = None
def get_db_password(db_name):
"""get db password from defs"""
import defs
if hasattr(defs, 'get_db_password'):
return defs.get_db_password(db_name)
elif hasattr(defs, 'db_password'):
return defs.db_password
else:
return db_name
whitelisted = []
guest_methods = []
def whitelist(allow_guest=False):
"""
decorator for whitelisting a function
Note: if the function is allowed to be accessed by a guest user,
it must explicitly be marked as allow_guest=True
"""
def innerfn(fn):
global whitelisted, guest_methods
whitelisted.append(fn)
if allow_guest:
guest_methods.append(fn)
return fn
return innerfn
def clear_cache(user=None):
"""clear boot cache"""
from webnotes.session_cache import clear
clear(user)
def get_roles():
"""get roles of current user"""
if session['user']=='Guest':
return ['Guest']
roles = [r[0] for r in conn.sql("""select distinct role from tabUserRole
where parent=%s and role!='All'""", session['user'])]
return roles + ['All']