fetch_instances.py

   1 import sqlite3
   2 import sys
   3 import json
   4 import fba
   5
   6 domain = sys.argv[1]
   7
   8 peerlist = fba.get_peers(domain)
   9
  10 if (peerlist is None):
  11     print("FATAL: CANNOT FETCH PEERS:", domain)
  12     sys.exit(255)
  13
  14 fba.c.execute(
  15     "SELECT domain FROM instances WHERE 1"
  16 )
  17
  18 for instance in peerlist:
  19     instance = instance.lower()
  20
  21     blacklisted = False
  22     for domain in fba.blacklist:
  23         if domain in instance:
  24             blacklisted = True
  25
  26     if blacklisted:
  27         print("WARNING: domain is blacklisted:", domain)
  28         continue
  29
  30     print("INFO: Handling instance:", instance)
  31     try:
  32         fba.c.execute(
  33             "SELECT domain FROM instances WHERE domain = ? LIMIT 1", (instance,)
  34         )
  35
  36         if fba.c.fetchone() == None:
  37             fba.add_instance(instance)
  38
  39         fba.conn.commit()
  40     except Exception as e:
  41         print("ERROR:", e, instance)
  42
  43 fba.conn.close()