examples/python/create-database.py

   1 #!/usr/bin/python3
   2
   3 import 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 an AS
  24     a = w.add_as(204867)
  25     a.name = "Lightning Wire Labs GmbH"
  26
  27     print(a)
  28
  29     # Add a network
  30     n = w.add_network("2a07:1c44:5800::/40")
  31     n.country_code = "DE"
  32     n.asn = a.number
  33
  34     print(n)
  35
  36     # Write the database to disk
  37     for f in sys.argv[1:]:
  38         w.write(f)