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")
