mongo_client = MongoClient(pymongo.Connection(settings.mongodb_host)) parser = OptionParser() parser.add_option("-r", "--reset_db", dest="reset_db", help="reset all database of this site(use with caution)", default="no") parser.add_option("-i", "--site_id", dest="site_id", help="Site ID(required)", default=None) parser.add_option("-n", "--site_name", dest="site_name", help="Site Name(required)", default=None) parser.add_option("-c", "--calc_interval", dest="calc_interval", help="How long (in seconds) the server will update the calculation", default="43200") (options, args) = parser.parse_args() site_id, site_name = options.site_id, options.site_name #assert site_id is not None #assert site_name is not None # TODO: use dropDatabase? connection = pymongo.Connection(settings.mongodb_host) if options.reset_db == "yes": getSiteDBCollection(connection, site_id, "item_similarities").drop() getSiteDBCollection(connection, site_id, "raw_logs").drop() getSiteDBCollection(connection, site_id, "items").drop() getSiteDB(connection, site_id).create_collection("raw_logs", {}) getSiteDBCollection(connection, site_id, "raw_logs").ensure_index([("timestamp", -1)]) mongo_client.updateSite(site_id, site_name, int(options.calc_interval))
parser.add_option("-n", "--site_name", dest="site_name", help="Site Name(required)", default=None) parser.add_option( "-c", "--calc_interval", dest="calc_interval", help="How long (in seconds) the server will update the calculation", default="43200") (options, args) = parser.parse_args() site_id, site_name = options.site_id, options.site_name #assert site_id is not None #assert site_name is not None # TODO: use dropDatabase? connection = pymongo.Connection(settings.mongodb_host) if options.reset_db == "yes": getSiteDBCollection(connection, site_id, "item_similarities").drop() getSiteDBCollection(connection, site_id, "raw_logs").drop() getSiteDBCollection(connection, site_id, "items").drop() getSiteDB(connection, site_id).create_collection("raw_logs", {}) getSiteDBCollection(connection, site_id, "raw_logs").ensure_index([("timestamp", -1)]) mongo_client.updateSite(site_id, site_name, int(options.calc_interval))