def add_team(query, verbose=False): assert query["name"] assert query["link"] if verbose: print "Do you want to add the team {\"name\": %s, \"link\": %s} [Y/n]:" % (query["name"], query["link"]) while 42: line = os.sys.stdin.readline() line = line[:-1] if line in ["Y", "y", "yes"]: break elif line in ["N", "n", "no"]: return else: print "[Y/n]:" old = find_one({"link": query["link"]}, "prono", "teams") if old is None: print "added team :", query["name"] return insert(query, "prono", "teams") else: if old["name"] != query["name"]: print old["name"] print query["name"] print # print ret # print "%s already inserted" % (query["name"]) return None
def add_learning_object(query, connection=None): assert query.get("name", None) if find_one({"name": query["name"]}, "prono", "learning_objects"): return "learning object named %s already exists" % (query["name"]) return insert(query, "prono", "learning_objects", connection)
def add_competition(query): assert query.get("name", None) assert query.get("region", None) if find_one({"name": query["name"]}, "prono", "competitions"): return "competition already exists" region = find_one({"name": query["region"]}, "prono", "regions") if region == None: add_region({"name": query["region"]}) region = find_one({"name": query["region"]}, "prono", "regions") region_id = region["_id"] return insert({ "name": query["name"], "region": region_id }, "prono", "competitions")
def load_from_db(self, name): if self.client_db is None: self.client_db = get_db("prono", mongolab=True) db_obj = find_one({"name": name}, "prono", "learning_objects", connection=self.client_db) self.name = db_obj["name"] self.params = db_obj["params"] self.nn = pickle.loads(db_obj["nn"]) self.scalizer = pickle.loads(db_obj["scalizer"]) self.normalizer = pickle.loads(db_obj["normalizer"])
def find_team(query, connection=None): return find_one(query, "prono", "teams", connection)
def find_game(query, connection=None): return find_one(query, "prono", "games", connection)
def find_learning_object_by_id(lo_id, connection=None): return find_one({"_id": ObjectId(lo_id)}, "prono", "learning_objects", connection)
def find_learning_object(query, connection=None): return find_one(query, "prono", "learning_objects", connection)
def find_stadium(query, connection=None): return find_one(query, "prono", "stadiums", connection)
def find_people(query): return find_one(query, "prono", "peoples")
def find_region(query, connection=None): return find_one(query, "prono", "regions", connection)
def find_competition_by_id(ch_id, connection=None): return find_one({"_id": ObjectId(ch_id)}, "prono", "competitions", connection)
def find_competition(query, connection=None): return find_one(query, "prono", "competitions", connection)
def find_season(query): return find_one(query, "prono", "seasons")