9 from wizard import app, install, resolve, sql, util
10 from wizard.app import php
12 def make_filename_regex_define(var):
13 """See :ref:`versioning config <seed>` for more information."""
14 return 'wp-config.php', php.re_define(var)
16 seed = util.dictmap(make_filename_regex_define, {
17 # these funny names are due to convention set by MediaWiki
18 'WIZARD_DBSERVER': 'DB_HOST',
19 'WIZARD_DBNAME': 'DB_NAME',
20 'WIZARD_DBUSER': 'DB_USER',
21 'WIZARD_DBPASSWORD': 'DB_PASSWORD',
22 'WIZARD_SECRETKEY': 'SECRET_KEY',
23 'WIZARD_AUTH_KEY': 'AUTH_KEY',
24 'WIZARD_SECURE_AUTH_KEY': 'SECURE_AUTH_KEY',
25 'WIZARD_LOGGED_IN_KEY': 'LOGGED_IN_KEY',
26 'WIZARD_NONCE_KEY': 'NONCE_KEY',
29 class Application(app.Application):
31 parametrized_files = ['wp-config.php'] + php.parametrized_files
32 extractors = app.make_extractors(seed)
33 extractors.update(php.extractors)
34 substitutions = app.make_substitutions(seed)
35 substitutions.update(php.substitutions)
36 install_schema = install.ArgSchema("db", "admin", "email", "title")
37 deprecated_keys = set(['WIZARD_SECRETKEY'])
38 random_keys = set(['WIZARD_SECRETKEY', 'WIZARD_AUTH_KEY', 'WIZARD_SECURE_AUTH_KEY', 'WIZARD_LOGGED_IN_KEY', 'WIZARD_NONCE_KEY'])
39 def urlFromExtract(self, deployment):
41 meta = sql.connect(deployment.dsn)
42 wp_options = meta.tables["wp_options"]
43 query = wp_options.select(wp_options.c.option_name == 'home')
44 return query.execute().fetchone()['option_value']
45 except sqlalchemy.exc.OperationalError:
47 def download(self, version):
48 return "http://wordpress.org/wordpress-%s.tar.gz" % version
49 def checkConfig(self, deployment):
50 return os.path.isfile("wp-config.php")
51 def checkWeb(self, deployment):
52 # XXX: this sucks pretty hard
54 return self.checkWebPage(deployment, "",
55 outputs=["<html", "WordPress", "feed"],
56 exclude=["Error establishing a database connection"])
58 deployment.enableOldStyleUrls()
62 def detectVersion(self, deployment):
63 return self.detectVersionFromFile("wp-includes/version.php", php.re_var("wp_version"))
64 def install(self, version, options):
65 util.soft_unlink("wp-config.php")
68 'dbhost': options.dsn.host,
69 'uname': options.dsn.username,
70 'dbname': options.dsn.database,
71 'pwd': options.dsn.password,
77 'weblog_title': options.title,
78 'admin_email': options.email,
82 old_mode = os.stat(".").st_mode
83 os.chmod(".", 0777) # XXX: squick squick
85 # we need to disable the wp_mail function in wp-includes/pluggable[-functions].php
86 pluggable_path = os.path.exists('wp-includes/pluggable.php') and 'wp-includes/pluggable.php' or 'wp-includes/pluggable-functions.php'
87 pluggable = open(pluggable_path, 'r').read()
88 wp_mail_noop = "<?php function wp_mail( $to, $subject, $message, $headers = '', $attachments = array() ) { /*noop*/ } ?> \n\n"
89 pluggable_file = open(pluggable_path,'w')
90 pluggable_file.write(wp_mail_noop)
91 pluggable_file.write(pluggable)
92 pluggable_file.close()
94 result = install.fetch(options, "wp-admin/setup-config.php?step=2", post_setup_config)
95 logging.debug("setup-config.php output\n\n" + result)
96 result = install.fetch(options, "wp-admin/install.php?step=2", post_install)
97 logging.debug("install.php output\n\n" + result)
98 os.chmod(".", old_mode)
99 if "Finished" not in result and "Success" not in result:
100 raise app.InstallFailure()
102 # not sure what to do about this
103 meta = sql.connect(options.dsn)
104 wp_options = meta.tables["wp_options"]
105 wp_options.update().where(wp_options.c.option_name == 'siteurl').values(option_value=options.web_path).execute()
106 wp_options.update().where(wp_options.c.option_name == 'home').values(option_value="http://%s%s" % (options.web_host, options.web_path)).execute() # XXX: what if missing leading slash; this should be put in a function
107 # should also set the username and admin password
109 wp_users = meta.tables["wp_users"]
110 hashed_pass = hashlib.md5(options.admin_password).hexdigest()
111 wp_users.update().where(wp_users.c.ID == 1).values(user_login=options.admin_name,user_nicename=options.admin_name,display_name=options.admin_name,user_pass=hashed_pass).execute()
112 wp_usermeta = meta.tables["wp_usermeta"]
113 wp_usermeta.delete().where(wp_usermeta.c.user_id==1 and wp_usermeta.c.meta_key == "default_password_nag").execute()
115 # now we can restore the wp_mail function in wp-includes/pluggable[-functions].php
116 pluggable_file = open(pluggable_path,'w')
117 pluggable_file.write(pluggable)
118 pluggable_file.close()
120 php.ini_replace_vars()
121 def upgrade(self, d, version, options):
122 result = d.fetch("wp-admin/upgrade.php?step=1")
123 if "Upgrade Complete" not in result and "No Upgrade Required" not in result:
124 raise app.UpgradeFailure(result)
125 def backup(self, deployment, backup_dir, options):
126 app.backup_database(backup_dir, deployment)
127 def restore(self, deployment, backup_dir, options):
128 app.restore_database(backup_dir, deployment)
129 def remove(self, deployment, options):
130 app.remove_database(deployment)
132 Application.resolutions = {
137 /** WordPress absolute path to the Wordpress directory. */
139 /** WordPress absolute path to the Wordpress directory. */
141 /** Absolute path to the WordPress directory. */