tools: Import the copy script

This is useful to debug the library by copying one library 1:1 and then
diffing the result.

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/Makefile.am b/Makefile.am
index e41ae8386de6b9352e98836f2b34269661ab5ccc..d4305507c8b6a5f8cd31911472c2caaec8e1af48 100644 (file)
--- a/Makefile.am
+++ b/Makefile.am
@@ -541,6 +541,9 @@ man/%.html: man/%.txt man/asciidoc.conf
 upload-man: $(MANPAGES_HTML)
        rsync -avHz --delete --progress $(MANPAGES_HTML) ms@fs01.haj.ipfire.org:/pub/man-pages/$(PACKAGE_NAME)/
 
+EXTRA_DIST += \
+       tools/copy.py
+
 EXTRA_DIST += \
        debian/build.sh \
        debian/changelog \

diff --git a/po/POTFILES.in b/po/POTFILES.in
index da20831c3bbf1ab87135c979f7856e36411dc662..57391d222fecfca3c64a2bb337c05e26fd8596e0 100644 (file)
--- a/po/POTFILES.in
+++ b/po/POTFILES.in
@@ -11,3 +11,4 @@ src/scripts/location-importer.in
 src/scripts/location.in
 src/systemd/location-update.service.in
 src/systemd/location-update.timer.in
+tools/copy.py

diff --git a/tools/copy.py b/tools/copy.py
new file mode 100644 (file)
index 0000000..39129c0
--- /dev/null
+++ b/tools/copy.py
@@ -0,0 +1,108 @@
+#!/usr/bin/python3
+###############################################################################
+#                                                                             #
+# libloc - A library to determine the location of someone on the Internet     #
+#                                                                             #
+# Copyright (C) 2024 IPFire Development Team <info@ipfire.org>                #
+#                                                                             #
+# This library is free software; you can redistribute it and/or               #
+# modify it under the terms of the GNU Lesser General Public                  #
+# License as published by the Free Software Foundation; either                #
+# version 2.1 of the License, or (at your option) any later version.          #
+#                                                                             #
+# This library is distributed in the hope that it will be useful,             #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of              #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU           #
+# Lesser General Public License for more details.                             #
+#                                                                             #
+###############################################################################
+
+import argparse
+
+import location
+from location.i18n import _
+
+flags = (
+       location.NETWORK_FLAG_ANONYMOUS_PROXY,
+       location.NETWORK_FLAG_SATELLITE_PROVIDER,
+       location.NETWORK_FLAG_ANYCAST,
+       location.NETWORK_FLAG_DROP,
+)
+
+def copy_all(db, writer):
+       # Copy vendor
+       if db.vendor:
+               writer.vendor = db.vendor
+
+       # Copy description
+       if db.description:
+               writer.description = db.description
+
+       # Copy license
+       if db.license:
+               writer.license = db.license
+
+       # Copy all ASes
+       for old in db.ases:
+               new = writer.add_as(old.number)
+               new.name = old.name
+
+       # Copy all networks
+       for old in db.networks:
+               new = writer.add_network("%s" % old)
+
+               # Copy country code
+               new.country_code = old.country_code
+
+               # Copy ASN
+               if old.asn:
+                       new.asn = old.asn
+
+               # Copy flags
+               for flag in flags:
+                       if old.has_flag(flag):
+                               new.set_flag(flag)
+
+       # Copy countries
+       for old in db.countries:
+               new = writer.add_country(old.code)
+
+               # Copy continent code
+               new.continent_code = old.continent_code
+
+               # Copy name
+               new.name = old.name
+
+def main():
+       """
+               Main Function
+       """
+       parser = argparse.ArgumentParser(
+               description=_("Copies a location database"),
+       )
+
+       # Input File
+       parser.add_argument("input-file", help=_("File to read"))
+
+       # Output File
+       parser.add_argument("output-file", help=_("File to write"))
+
+       # Parse arguments
+       args = parser.parse_args()
+
+       input_file  = getattr(args, "input-file")
+       output_file = getattr(args, "output-file")
+
+       # Open the database
+       db = location.Database(input_file)
+
+       # Create a new writer
+       writer = location.Writer()
+
+       # Copy everything
+       copy_all(db, writer)
+
+       # Write the new file
+       writer.write(output_file)
+
+main()