seitime-frappe/cgi-bin/webnotes/install_lib/install.py
2011-06-27 11:23:28 +05:30

245 lines
6.5 KiB
Python
Executable file

import os,sys
cgi_bin_path = os.path.sep.join(__file__.split(os.path.sep)[:-3])
sys.path.append(cgi_bin_path)
#
# make a copy of defs.py (if not exists)
#
def copy_defs():
global cgi_bin_path
if not os.path.exists(os.path.join(cgi_bin_path, 'webnotes', 'defs.py')):
ret = os.system('cp '+ os.path.join(cgi_bin_path, 'webnotes', 'defs_template.py')+\
' '+os.path.join(cgi_bin_path, 'webnotes', 'defs.py'))
print 'Made copy of defs.py'
#
# Main Installer Class
#
class Installer:
def __init__(self, root_login, root_password):
import webnotes
import webnotes.db
import webnotes.defs
self.root_password = root_password
from webnotes.model.db_schema import DbManager
self.conn = webnotes.db.Database(user=root_login, password=root_password)
webnotes.conn=self.conn
webnotes.session= {'user':'Administrator'}
self.dbman = DbManager(self.conn)
self.mysql_path = hasattr(webnotes.defs, 'mysql_path') and webnotes.defs.mysql_path or ''
#
# run framework related cleanups
#
def framework_cleanups(self, target):
import webnotes
self.dbman.drop_table('__DocTypeCache')
webnotes.conn.sql("create table `__DocTypeCache` (name VARCHAR(120), modified DATETIME, content TEXT, server_code_compiled TEXT)")
# set the basic passwords
webnotes.conn.begin()
webnotes.conn.sql("update tabProfile set password = password('admin') where name='Administrator'")
webnotes.conn.commit()
def import_core_module(self):
"""
Imports the "Core" module from .txt file and creates
Creates profile Administrator
"""
import webnotes
from webnotes.modules.import_module import import_module
from webnotes.modules.module_manager import reload_doc
reload_doc('core','doctype','doctype')
reload_doc('core','doctype','docfield')
reload_doc('core','doctype','docperm')
import_module('core')
webnotes.conn.begin()
from webnotes.model.doc import Document
p = Document('Profile')
p.name = p.first_name = 'Administrator'
p.email = 'admin@localhost'
p.save(new = 1)
ur = Document('UserRole')
ur.parent = 'Administrator'
ur.role = 'Administrator'
ur.parenttype = 'Profile'
ur.parentfield = 'userroles'
p.enabled = 1
ur.save(1)
p = Document('Profile')
p.name = p.first_name = 'Guest'
p.email = 'guest@localhost'
p.enabled = 1
p.save(new = 1)
ur = Document('UserRole')
ur.parent = 'Guest'
ur.role = 'Guest'
ur.parenttype = 'Profile'
ur.parentfield = 'userroles'
ur.save(1)
webnotes.conn.commit()
#
# main script to create a database from
#
def import_from_db(self, target, source_path='', password = 'admin', verbose=0):
"""
a very simplified version, just for the time being..will eventually be deprecated once the framework stabilizes.
"""
import webnotes.defs
#Storing passed source path
passed_source_path = source_path
# get the path of the sql file to import
if not source_path:
source_path = os.path.join(os.path.sep.join(os.path.abspath(webnotes.__file__).split(os.path.sep)[:-3]), 'data', 'Framework.sql')
# delete user (if exists)
self.dbman.delete_user(target)
# create user and db
self.dbman.create_user(target,getattr(webnotes.defs,'db_password',None))
if verbose: print "Created user %s" % target
# create a database
self.dbman.create_database(target)
if verbose: print "Created database %s" % target
# grant privileges to user
self.dbman.grant_all_privileges(target,target)
if verbose: print "Granted privileges to user %s and database %s" % (target, target)
# flush user privileges
self.dbman.flush_privileges()
self.conn.use(target)
# import in target
if verbose: print "Starting database import..."
self.dbman.restore_database(target, source_path, self.root_password.replace(' ','\ '))
if verbose: print "Imported from database %s" % source_path
#If source path is passed
#i.e. importing from master sql, dont import core modules
if not passed_source_path: self.import_core_module()
# framework cleanups
self.framework_cleanups(target)
if verbose: print "Ran framework startups on %s" % target
return target
def make_scheduler(root_login, root_password, verbose):
"""
Make the database where all scheduler events will be stored from multiple datbases
See webnotes.utils.scheduler for more information
"""
conn = webnotes.db.Database(user=root_login, password=root_password)
from webnotes.model.db_schema import DbManager
dbman = DbManager(conn)
# delete user (if exists)
dbman.delete_user('master_scheduler')
# create user and db
dbman.create_user('master_scheduler', getattr(webnotes.defs,'db_password',None))
if verbose: print "Created user master_scheduler"
# create a database
dbman.create_database('master_scheduler')
if verbose: print "Created database master_scheduler"
# grant privileges to user
dbman.grant_all_privileges('master_scheduler','master_scheduler')
# flush user privileges
dbman.flush_privileges()
conn.use('master_scheduler')
# create events table
conn.sql("""create table Event(
`db_name` varchar(60),
`event` varchar(180),
`interval` int(20),
`next_execution` timestamp,
`recurring` int(1),
primary key (`db_name`, `event`),
index next_execution(next_execution)
)""")
conn.sql("""create table EventLog(
`db_name` varchar(180),
`event` varchar(180),
`executed_on` timestamp,
`log` text,
index executed_on(executed_on))
""")
#
# load the options
#
def get_parser():
from optparse import OptionParser
parser = OptionParser(usage="usage: %prog [options] ROOT_LOGIN ROOT_PASSWORD DBNAME")
parser.add_option("-x", "--database-password", dest="password", default="admin", help="Optional: New password for the Framework Administrator, default 'admin'")
parser.add_option("-s", "--source", dest="source_path", default=None, help="Optional: Path of the sql file from which you want to import the instance, default 'data/Framework.sql'")
return parser
#
# execution here
#
if __name__=='__main__':
parser = get_parser()
(options, args) = parser.parse_args()
try:
import webnotes
import webnotes.db
import webnotes.defs
except ImportError:
copy_defs()
import webnotes
import webnotes.db
import webnotes.defs
if len(args)==3:
root_login, root_password, db_name = args[0], args[1], args[2]
if db_name=='master_scheduler':
make_scheduler(root_login, root_password, 1)
else:
inst = Installer(root_login, root_password)
inst.import_from_db(db_name, source_path=options.source_path, \
password = options.password, verbose = 1)
print "Database created, please edit defs.py to get started"
else:
parser.print_help()