examples/python/create-database.py

   1 #!/usr/bin/python3
   2
   3 import _location as location
   4 import os
   5 import sys
   6
   7 ABS_SRCDIR = os.environ.get("ABS_SRCDIR", ".")
   8
   9 private_key_path = os.path.join(ABS_SRCDIR, "examples/private-key.pem")
  10
  11 with open(private_key_path, "r") as pkey:
  12     w = location.Writer(pkey)
  13
  14     # Set the vendor
  15     w.vendor = "IPFire Project"
  16
  17     # Set a description
  18     w.description = "This is a geo location database"
  19
  20     # Set a license
  21     w.license = "CC"
  22
  23     # Add a country
  24     c = w.add_country("DE")
  25
  26     # Add an AS
  27     a = w.add_as(204867)
  28     a.name = "Lightning Wire Labs GmbH"
  29
  30     print(a)
  31
  32     # Add a network
  33     n = w.add_network("2a07:1c44:5800::/40")
  34     n.country_code = "DE"
  35     n.asn = a.number
  36
  37     print(n)
  38
  39     # Write the database to disk
  40     for f in sys.argv[1:]:
  41         w.write(f)