コード例 #1
0
ファイル: update_site.py プロジェクト: jacobfan/poco
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))

コード例 #2
0
ファイル: update_site.py プロジェクト: stipid/poco
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))