def main():
# Parse options
p = optparse.OptionParser()
p.add_option('--database', '-b', action="store", type="string",
help="Database management. Values: "
"create - create new DB, "
"update - update existing DB without losing the data")
p.add_option('--update-version', '-u', default="all", type="string",
help="Valid only for updating via HTTP. "
"Version update numbers for processing. Can be 111 or 111-222 or 111,222,333."
"For '--database-create' only one value (like 111) may be specified. If not specified, "
"all updates will be processed (for '--database update') or last DB snapshot "
"(for '--database create')")
p.add_option('--show-versions', '-v', action="store_true", dest="show_versions", default=False,
help="Show available fias versions. "
"These version numbers are required for the '--update-version' option")
p.add_option('--source', '-s', default="http",
help="Create/update DB from source. Value: 'http', absolute path to folder containing XMLs "
"or absolute path to rar file.")
p.add_option('--sphinx-configure', '-c', action="store_true", dest="sphinx", default="False",
help="Configure Sphinx. Creates a sphinx.conf file specified in '--output-conf'")
p.add_option('--indexer-path', '-i',
help="Path to Sphinx indexer binary. Required for '--sphinx-configure'")
p.add_option('--output-conf', '-o',
help="Output config filename. Required for '--sphinx-configure'")
options, arguments = p.parse_args()
# if no arguments
if len(sys.argv) < 2:
print("Py-Phias manager. Try manage.py --help for options.")
return
# Show FIAS updates
if options.show_versions:
print_fias_versions()
return
# Manage DB
if options.database:
# create new database
aoupdater = Updater(options.source)
allowed_updates = None
if options.source == "http":
allowed_updates = get_allowed_updates(options.update_version, options.database)
if options.database == "create":
aoupdater.create(allowed_updates)
# update database
if options.database == "update":
aoupdater.update(allowed_updates)
# Manage Sphinx
if options.sphinx and options.indexer_path and options.output_conf:
sphinxh = SphinxHelper()
sphinxh.configure_indexer(options.indexer_path, options.output_conf)
def print_fias_versions():
imp = SoapReceiver()
current_version = Updater.get_current_fias_version()
all_versions = imp.get_update_list()
print("Installed version: {}".format(current_version))
print("Avaliable updates:")
print("Number\t\tDate")
for upd in all_versions:
mark_current = (' ', '*')[int(upd['intver']) == current_version]
print("{}{}\t\t{}".format(mark_current, upd['intver'], upd['strver']))
def get_allowed_updates(updates_str, mode="create"):
imp = SoapReceiver()
current_version = Updater.get_current_fias_version()
all_versions = [x for x in imp.get_update_list()]
user_defined_list = parse_update_str(updates_str)
out_list = []
if mode == "create":
if not user_defined_list:
yield all_versions[-1]
else:
assert len(user_defined_list) == 1, "Ony single update number allowed for DB create"
if mode == "update":
for uv in all_versions:
uv_ver = uv['intver']
if uv_ver > current_version and (not user_defined_list or uv_ver in user_defined_list):
out_list.append(uv)
out_list.sort(key=lambda item: item['intver'])
for ol_entry in out_list:
yield ol_entry
