squert_ip2c_update.py 文件源码

def main():
  try:
    import mysql.connector
  except:
    import sys
    print "\n[IP2C: WARNING]"
    print 'ERROR: mysql.connector library not available. Exitting...'
    sys.exit(1)
  import os
  args = parse_arguments()
  try:
    csv_file = os.path.join(os.environ['tmp'], 'ip2c-results.csv')
  except:
    csv_file = os.path.join('/tmp', 'ip2c-results.csv')
  ##create_table("securityonion_db", "ip2c")
  print "\n[IP2C: Parse RIR DBs]"
  create_tmp_file(args.source_dir, csv_file)
  print "\n[IP2C: Update MySQL DB]"
  update_table("securityonion_db", "ip2c", csv_file)
  print "\n[IP2C: Update Validation]"
  read_table("securityonion_db", "ip2c")