113 lines
2.5 KiB
Python
113 lines
2.5 KiB
Python
import argparse
|
|
import os
|
|
import sys
|
|
import unpacker
|
|
import setup
|
|
|
|
'''
|
|
obtain new info for sites database
|
|
install core files
|
|
move wp-content to correct place
|
|
import db to website
|
|
check for any issues
|
|
|
|
package.json
|
|
|
|
function ai1wm_activate_plugins( $plugins ) {
|
|
$current = get_option( AI1WM_ACTIVE_PLUGINS, array() );
|
|
|
|
// Add plugins
|
|
foreach ( $plugins as $plugin ) {
|
|
if ( ! in_array( $plugin, $current ) && ! is_wp_error( validate_plugin( $plugin ) ) ) {
|
|
$current[] = $plugin;
|
|
}
|
|
}
|
|
|
|
return update_option( AI1WM_ACTIVE_PLUGINS, $current );
|
|
}
|
|
'''
|
|
|
|
def setup_args():
|
|
parser = argparse.ArgumentParser(
|
|
prog='AiO-Restore',
|
|
description='Unpack and Restore All-in-One WP Migration Packages',
|
|
epilog='aio_restore ',
|
|
)
|
|
parser.add_argument(
|
|
'source',
|
|
type=str,
|
|
help='Source Path'
|
|
)
|
|
parser.add_argument(
|
|
'target',
|
|
type=str,
|
|
help='Target Path'
|
|
)
|
|
parser.add_argument(
|
|
'db_name',
|
|
type=str,
|
|
help='Database Name'
|
|
)
|
|
parser.add_argument(
|
|
'db_pass',
|
|
type=str,
|
|
help='Database Password'
|
|
)
|
|
parser.add_argument(
|
|
name='site_url',
|
|
name='-s',
|
|
name='--site',
|
|
type=str,
|
|
help='Site URL, domain only'
|
|
)
|
|
parser.add_argument(
|
|
'-h',
|
|
'--home',
|
|
'home_url',
|
|
type=str,
|
|
help='Home URL, domain only'
|
|
)
|
|
parser.add_argument(
|
|
'-v',
|
|
'--version',
|
|
action='version',
|
|
version='%(prog)s v0.0.1',
|
|
help='Show version information'
|
|
)
|
|
|
|
return parser
|
|
|
|
plugin_prefix = 'SERVMASK_PREFIX_'
|
|
|
|
if __name__ == '__main__':
|
|
""" Entry of the ai1wm program. """
|
|
|
|
parser = setup_args()
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
# Ensure we have source and target
|
|
if not args.source or not args.target:
|
|
parser.print_help()
|
|
exit(1)
|
|
|
|
# Unpack the archive and files to the source destination
|
|
if os.path.isfile(args.source):
|
|
unpacker.Ai1wmUnpacker().unpack(args.source, args.target)
|
|
except Exception as e:
|
|
print(e)
|
|
sys.exit(-1)
|
|
|
|
# Process WordPress setup and install
|
|
try:
|
|
# TODO: check for package.json for processing or exit
|
|
if os.path.isfile(args.target):
|
|
# check for database.sql and package.json
|
|
# Setup Site files
|
|
setup.WPSetup()
|
|
except Exception as e:
|
|
print(e)
|
|
sys.exit(-1)
|
|
|
|
|
|
sys.exit(0) |