update files
This commit is contained in:
parent
535a217a47
commit
bc16aae3b4
6
error.py
6
error.py
@ -1,6 +0,0 @@
|
||||
""" Package specific exceptions. """
|
||||
|
||||
|
||||
class Ai1wmError(Exception):
|
||||
""" Exceptions raised from this package. """
|
||||
pass
|
||||
120
header.py
120
header.py
@ -1,120 +0,0 @@
|
||||
""" Parses ai1wm file header. """
|
||||
|
||||
import collections
|
||||
import struct
|
||||
from error import Ai1wmError
|
||||
from tools import b__, s__
|
||||
|
||||
|
||||
class Ai1wmHeader(tuple):
|
||||
""" Parses an `All-in-One WP Migration` header. """
|
||||
|
||||
SIZE = 4377
|
||||
EOF = b'\x00' * SIZE
|
||||
|
||||
_Location = collections.namedtuple('_Location', ['offset', 'size'])
|
||||
_LOC_NAME = _Location(0, 255) # File name
|
||||
_LOC_SIZE = _Location(255, 14) # File size
|
||||
_LOC_TIME = _Location(269, 12) # Last modified time
|
||||
_LOC_PATH = _Location(281, 4096) # File path
|
||||
|
||||
def __new__(cls, path=None, name=None, size=None, time=None):
|
||||
""" Returns a new instance of the object. """
|
||||
|
||||
if path or name or size or time:
|
||||
if not isinstance(path, str) or path == '':
|
||||
raise ValueError('<path> must be a nonempty string')
|
||||
if not isinstance(name, str) or name == '':
|
||||
raise ValueError('<name> must be a nonempty string')
|
||||
if not isinstance(size, int) or size < 0:
|
||||
raise ValueError('<size> must be a non-negative integer')
|
||||
if not isinstance(time, int) or time < 0:
|
||||
raise ValueError('<time> must be a non-negative integer')
|
||||
|
||||
return super(Ai1wmHeader, cls).__new__(cls, [path, name, size, time])
|
||||
|
||||
@classmethod
|
||||
def unpack(cls, header):
|
||||
""" Unpacks a binary header. """
|
||||
|
||||
if len(header) != cls.SIZE:
|
||||
raise Ai1wmError('invalid header size')
|
||||
|
||||
if header == cls.EOF:
|
||||
return cls()
|
||||
|
||||
return cls(
|
||||
path=s__(cls.__extract_field(header, cls._LOC_PATH)),
|
||||
name=s__(cls.__extract_field(header, cls._LOC_NAME)),
|
||||
size=cls.__extract_int(header, cls._LOC_SIZE),
|
||||
time=cls.__extract_int(header, cls._LOC_TIME),
|
||||
)
|
||||
|
||||
def pack(self):
|
||||
""" Packs to a binary header. """
|
||||
|
||||
attributes, formats, locations = [], '', [
|
||||
('name', self._LOC_NAME),
|
||||
('size', self._LOC_SIZE),
|
||||
('time', self._LOC_TIME),
|
||||
('path', self._LOC_PATH),
|
||||
]
|
||||
|
||||
for name, location in locations:
|
||||
attribute = b__(getattr(self, name))
|
||||
if len(attribute) > location.size:
|
||||
raise Ai1wmError('{} is too long to pack: {}'.format(name, getattr(self, name)))
|
||||
attributes.append(attribute)
|
||||
formats += '{}s'.format(location.size)
|
||||
|
||||
return struct.pack(formats, *attributes)
|
||||
|
||||
@property
|
||||
def path(self):
|
||||
""" Path of the file. """
|
||||
|
||||
return self[0]
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
""" Name of the file. """
|
||||
|
||||
return self[1]
|
||||
|
||||
@property
|
||||
def size(self):
|
||||
""" Size of the file. """
|
||||
|
||||
return self[2]
|
||||
|
||||
@property
|
||||
def time(self):
|
||||
""" Time of the file. """
|
||||
|
||||
return self[3]
|
||||
|
||||
@property
|
||||
def eof(self):
|
||||
""" Indicates if this is an EOF header. """
|
||||
|
||||
return not any(self)
|
||||
|
||||
@classmethod
|
||||
def __extract_field(cls, header, location):
|
||||
""" Extracts a header field. """
|
||||
|
||||
try:
|
||||
field = struct.unpack_from('{}s'.format(location.size), header, offset=location.offset)[0]
|
||||
except struct.error as e:
|
||||
raise Ai1wmError('error extracting a header field, error: {}'.format(e))
|
||||
|
||||
return field.rstrip(b'\x00')
|
||||
|
||||
@classmethod
|
||||
def __extract_int(cls, header, location):
|
||||
""" Extracts an integral header field. """
|
||||
|
||||
try:
|
||||
return int(cls.__extract_field(header, location))
|
||||
except ValueError:
|
||||
raise Ai1wmError('invalid header field')
|
||||
18
main.py
18
main.py
@ -2,7 +2,6 @@ import argparse
|
||||
import os
|
||||
import sys
|
||||
import unpacker
|
||||
from error import Ai1wmError
|
||||
|
||||
'''
|
||||
obtain new info for sites database
|
||||
@ -10,6 +9,21 @@ 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 );
|
||||
}
|
||||
'''
|
||||
|
||||
if __name__ == '__main__':
|
||||
@ -23,7 +37,7 @@ if __name__ == '__main__':
|
||||
try:
|
||||
if os.path.isfile(args.source):
|
||||
unpacker.Ai1wmUnpacker().unpack(args.source, args.target)
|
||||
except Ai1wmError as e:
|
||||
except Exception as e:
|
||||
print(e)
|
||||
sys.exit(-1)
|
||||
|
||||
|
||||
36
tools.py
36
tools.py
@ -1,36 +0,0 @@
|
||||
import sys
|
||||
|
||||
|
||||
def s__(obj):
|
||||
"""
|
||||
Converts an object to str format.
|
||||
:rtype: str
|
||||
"""
|
||||
|
||||
if isinstance(obj, str):
|
||||
return obj
|
||||
v = sys.version_info[0]
|
||||
if v == 2:
|
||||
if isinstance(obj, unicode):
|
||||
return obj.encode('utf-8')
|
||||
elif v == 3:
|
||||
if isinstance(obj, bytes):
|
||||
return obj.decode('utf-8')
|
||||
return str(obj)
|
||||
|
||||
|
||||
def b__(obj):
|
||||
"""
|
||||
Converts an object to bytes format.
|
||||
:rtype: bytes
|
||||
"""
|
||||
|
||||
v = sys.version_info[0]
|
||||
if v == 2:
|
||||
return s__(obj)
|
||||
|
||||
if isinstance(obj, bytes):
|
||||
return obj
|
||||
if not isinstance(obj, str):
|
||||
obj = str(obj)
|
||||
return obj.encode('utf-8')
|
||||
163
unpacker.py
163
unpacker.py
@ -1,14 +1,159 @@
|
||||
""" Unpacks an `All-in-One WP Migration` package. """
|
||||
|
||||
import collections
|
||||
import errno
|
||||
import os
|
||||
from error import Ai1wmError
|
||||
from header import Ai1wmHeader
|
||||
from tools import s__
|
||||
import struct
|
||||
import sys
|
||||
|
||||
|
||||
class Ai1wmUnpacker(object):
|
||||
class Ai1wmUnpacker(tuple):
|
||||
""" Unpacks an `All-in-One WP Migration` package. """
|
||||
SIZE = 4377
|
||||
EOF = b'\x00' * SIZE
|
||||
|
||||
_Location = collections.namedtuple('_Location', ['offset', 'size'])
|
||||
_LOC_NAME = _Location(0, 255) # File name
|
||||
_LOC_SIZE = _Location(255, 14) # File size
|
||||
_LOC_TIME = _Location(269, 12) # Last modified time
|
||||
_LOC_PATH = _Location(281, 4096) # File path
|
||||
|
||||
def __new__(cls, path=None, name=None, size=None, time=None):
|
||||
""" Returns a new instance of the object. """
|
||||
|
||||
if path or name or size or time:
|
||||
if not isinstance(path, str) or path == '':
|
||||
raise ValueError('<path> must be a nonempty string')
|
||||
if not isinstance(name, str) or name == '':
|
||||
raise ValueError('<name> must be a nonempty string')
|
||||
if not isinstance(size, int) or size < 0:
|
||||
raise ValueError('<size> must be a non-negative integer')
|
||||
if not isinstance(time, int) or time < 0:
|
||||
raise ValueError('<time> must be a non-negative integer')
|
||||
|
||||
return super(Ai1wmUnpacker, cls).__new__(cls, [path, name, size, time])
|
||||
|
||||
@classmethod
|
||||
def unpack_header(cls, header):
|
||||
""" Unpacks a binary header. """
|
||||
|
||||
if len(header) != cls.SIZE:
|
||||
raise Exception('invalid header size')
|
||||
|
||||
if header == cls.EOF:
|
||||
return cls()
|
||||
|
||||
return cls(
|
||||
path=cls.s__(cls.__extract_field(header, cls._LOC_PATH)),
|
||||
name=cls.s__(cls.__extract_field(header, cls._LOC_NAME)),
|
||||
size=cls.__extract_int(header, cls._LOC_SIZE),
|
||||
time=cls.__extract_int(header, cls._LOC_TIME),
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def s__(cls, obj):
|
||||
"""
|
||||
Converts an object to str format.
|
||||
:rtype: str
|
||||
"""
|
||||
|
||||
if isinstance(obj, str):
|
||||
return obj
|
||||
v = sys.version_info[0]
|
||||
if v == 2:
|
||||
if isinstance(obj, unicode):
|
||||
return obj.encode('utf-8')
|
||||
elif v == 3:
|
||||
if isinstance(obj, bytes):
|
||||
return obj.decode('utf-8')
|
||||
return str(obj)
|
||||
|
||||
@classmethod
|
||||
def b__(cls, obj):
|
||||
"""
|
||||
Converts an object to bytes format.
|
||||
:rtype: bytes
|
||||
"""
|
||||
|
||||
v = sys.version_info[0]
|
||||
if v == 2:
|
||||
return cls.s__(obj)
|
||||
|
||||
if isinstance(obj, bytes):
|
||||
return obj
|
||||
if not isinstance(obj, str):
|
||||
obj = str(obj)
|
||||
return obj.encode('utf-8')
|
||||
|
||||
def pack(self):
|
||||
""" Packs to a binary header. """
|
||||
|
||||
attributes, formats, locations = [], '', [
|
||||
('name', self._LOC_NAME),
|
||||
('size', self._LOC_SIZE),
|
||||
('time', self._LOC_TIME),
|
||||
('path', self._LOC_PATH),
|
||||
]
|
||||
|
||||
for name, location in locations:
|
||||
attribute = self.b__(getattr(self, name))
|
||||
if len(attribute) > location.size:
|
||||
raise Exception('{} is too long to pack: {}'.format(name, getattr(self, name)))
|
||||
attributes.append(attribute)
|
||||
formats += '{}s'.format(location.size)
|
||||
|
||||
return struct.pack(formats, *attributes)
|
||||
|
||||
@property
|
||||
def path(self):
|
||||
""" Path of the file. """
|
||||
|
||||
return self[0]
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
""" Name of the file. """
|
||||
|
||||
return self[1]
|
||||
|
||||
@property
|
||||
def size(self):
|
||||
""" Size of the file. """
|
||||
|
||||
return self[2]
|
||||
|
||||
@property
|
||||
def time(self):
|
||||
""" Time of the file. """
|
||||
|
||||
return self[3]
|
||||
|
||||
@property
|
||||
def eof(self):
|
||||
""" Indicates if this is an EOF header. """
|
||||
|
||||
return not any(self)
|
||||
|
||||
@classmethod
|
||||
def __extract_field(cls, header, location):
|
||||
""" Extracts a header field. """
|
||||
|
||||
try:
|
||||
field = struct.unpack_from('{}s'.format(location.size), header, offset=location.offset)[0]
|
||||
except struct.error as e:
|
||||
raise Exception('error extracting a header field, error: {}'.format(e))
|
||||
|
||||
return field.rstrip(b'\x00')
|
||||
|
||||
@classmethod
|
||||
def __extract_int(cls, header, location):
|
||||
""" Extracts an integral header field. """
|
||||
|
||||
try:
|
||||
return int(cls.__extract_field(header, location))
|
||||
except ValueError:
|
||||
raise Exception('invalid header field')
|
||||
|
||||
|
||||
@staticmethod
|
||||
def __make_dirs(path, mode=0o777):
|
||||
@ -18,7 +163,7 @@ class Ai1wmUnpacker(object):
|
||||
os.makedirs(path, mode=mode)
|
||||
except OSError as e:
|
||||
if e.errno != errno.EEXIST:
|
||||
raise Ai1wmError('error creating a directory: {}, error: {}'.format(path, e))
|
||||
raise Exception('error creating a directory: {}, error: {}'.format(path, e))
|
||||
return path
|
||||
|
||||
@classmethod
|
||||
@ -33,7 +178,7 @@ class Ai1wmUnpacker(object):
|
||||
block_size = size
|
||||
block = stream.read(block_size)
|
||||
if len(block) != block_size:
|
||||
raise Ai1wmError('error extracting a file: {}, error: bad file size'.format(path))
|
||||
raise Exception('error extracting a file: {}, error: bad file size'.format(path))
|
||||
f.write(block)
|
||||
size -= len(block)
|
||||
|
||||
@ -42,7 +187,7 @@ class Ai1wmUnpacker(object):
|
||||
""" Unpacks a package. """
|
||||
|
||||
while True:
|
||||
header = Ai1wmHeader.unpack(stream.read(Ai1wmHeader.SIZE))
|
||||
header = cls.unpack_header(stream.read(Ai1wmUnpacker.SIZE))
|
||||
if header.eof:
|
||||
break
|
||||
|
||||
@ -56,10 +201,10 @@ class Ai1wmUnpacker(object):
|
||||
def unpack(cls, source, target):
|
||||
""" Unpacks a package. """
|
||||
|
||||
source, target = s__(os.path.realpath(source)), s__(os.path.realpath(target))
|
||||
source, target = cls.s__(os.path.realpath(source)), cls.s__(os.path.realpath(target))
|
||||
try:
|
||||
with open(source, 'rb') as f:
|
||||
cls.__unpack(f, target)
|
||||
except Exception as ex:
|
||||
raise Ai1wmError('error unpacking a file: {}, error: {}'.format(source, ex))
|
||||
raise Exception('error unpacking a file: {}, error: {}'.format(source, ex))
|
||||
return target
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user