def __init__(self, parms):
config = tmutil.parseWPConfig(
open(os.path.join(parms.wpdir, 'wp-config.php'), 'r'))
if parms.uselocal:
config['DB_HOST'] = 'localhost'
# Connect to the WP Database
self.conn = dbconn.dbconn(config['DB_HOST'], config['DB_USER'],
config['DB_PASSWORD'], config['DB_NAME'])
self.curs = self.conn.cursor()
self.prefix = config['table_prefix']
self.poststable = self.prefix + 'posts'
self.optionstable = self.prefix + 'options'
self.postmeta = self.prefix + 'postmeta'
self.venue_numbers = set()
parms.end = parms.now
if parms.now.month >= 5 and parms.now.month <= 9:
parms.start = parms.start.replace(month=6, day=1)
parms.end = parms.end.replace(month=9, day=30)
else:
parms.start = parms.start.replace(month=11, day=1)
parms.end = parms.end.replace(month=3, day=31)
if parms.now.month >= 10:
parms.end = parms.end.replace(year=parms.end.year + 1)
else:
parms.start = parms.start.replace(year=parms.start.year - 1)
# But we don't care about past trainings, set start to today
parms.start = parms.now.replace(hour=0, minute=0, second=0)
# Parse the configuration file
config = tmutil.parseWPConfig(
open(os.path.join(parms.wpdir, 'wp-config.php'), 'r'))
if parms.uselocal:
config['DB_HOST'] = 'localhost'
# Connect to the WP database
conn = dbconn.dbconn(config['DB_HOST'], config['DB_USER'],
config['DB_PASSWORD'], config['DB_NAME'])
curs = conn.cursor()
prefix = config['table_prefix']
poststable = prefix + 'posts'
optionstable = prefix + 'options'
# Find the taxonomy value for 'training'
stmt = "SELECT term_id FROM %s WHERE slug = 'training'" % (prefix +
'terms')
curs.execute(stmt)
# Establish parameters
parms = tmparms.tmparms()
# Add other parameters here
parms.add_argument('--configfile', type=str, default='/var/www/html/wp-config.php')
# Do global setup
globals.setup(parms,connect=False)
# Parse the current user's .my.cnf file; this must have the username and password in the [client] section.
msconfig = configparser.ConfigParser()
msconfig.read(os.path.expanduser("~/.my.cnf"))
username = msconfig.get("client", "user")
password = msconfig.get("client", "password")
# Parse the WordPress configuration file
config = tmutil.parseWPConfig(open(parms.configfile,'r'))
# Connect as the active user, who needs full authority over WordPress
conn = dbconn.dbconn('localhost', username, password, '')
curs = conn.cursor()
# Make sure the WP user can access the database
curs.execute("CREATE USER IF NOT EXISTS '%s'@'localhost' IDENTIFIED BY '%s'" % (config['DB_USER'], config['DB_PASSWORD']))
("GRANT ALL PRIVILEGES ON '%s'.'*' TO '%s'@'localhost'" % (config['DB_NAME'], config['DB_USER']))
curs.execute("FLUSH PRIVILEGES")
conn.close()
# Connect to the WP database
conn = dbconn.dbconn(config['DB_HOST'], config['DB_USER'], config['DB_PASSWORD'], config['DB_NAME'])
### Insert classes and functions here. The main program begins in the "if" statement below.
if __name__ == "__main__":
import tmparms
# Handle parameters
parms = tmparms.tmparms()
parms.add_argument('--quiet', '-q', action='count')
parms.add_argument('--verbose', '-v', action='count')
parms.add_argument('--wpconfigfile', type=str, default='/dev/null')
# Add other parameters here
parms.parse()
# Parse the WP Configuration file to find the database
config = tmutil.parseWPConfig(open(parms.wpconfigfile, 'r'))
# Connect to the database
conn = dbconn.dbconn(config['DB_HOST'], config['DB_USER'],
config['DB_PASSWORD'], config['DB_NAME'])
curs = conn.cursor()
prefix = config['table_prefix']
optiontable = prefix + 'options'
curs.execute("SELECT option_value FROM %s WHERE option_name = 'et_divi'" %
optiontable)
searchText = curs.fetchone()[0]
pattern = re.compile(
r"(</script>|^)(\s*)(<script>.*?GoogleAnalyticsObject.*?</script>)",
re.I | re.M | re.S)
m = re.search(pattern, searchText)
if not m: