4. Feladat

service.py

@@ -157,44 +157,97 @@ def vehicles():
 		conn.close()
 
 def __get_nevnap(date):
+	try:
 		resp = requests.get("https://db.bme.hu/~marton/shared/bin/nameday/nameday.cgi?month=%d&day=%d&format=json" % (date.month, date.day)).text
+	except:
+		return None
 	logging.debug(resp)
 	resp = json.loads(resp)
 	return resp["names"]
 
-@app.route('/jarmuvek/<jarmu_kulcs>.json')
-def show_vehicle(jarmu_kulcs):
+def __get_vehicle(jarmu_kulcs):
 	conn = get_db()
-	try:
 	try:
 		cur = conn.cursor()
+		try:
+			logging.debug("Getting data for vehicle with id %d" % jarmu_kulcs)
 			cur.execute('select telephely, berelheto, rendszam, alvazszam, uzembehelyezes, muszaki_ervenyesseg, marka, tipus, szin, teljesitmeny, ulesek, legkondicionalo from jarmu where jarmu_kulcs= :jk', jk=jarmu_kulcs)
 			res = cur.fetchone()
 			if(res == None):
-				abort(404)
+				return None
 			# else
 			telephely, berelheto, rendszam, alvazszam, uzembehelyezes, muszaki_ervenyesseg, marka, tipus, szin, teljesitmeny, ulesek, legkondicionalo = res
+			nevnap = __get_nevnap(uzembehelyezes)
+			if(uzembehelyezes != None):
+				uzembehelyezes = uzembehelyezes.isoformat()
+			if(muszaki_ervenyesseg != None):
+				muszaki_ervenyesseg = muszaki_ervenyesseg.isoformat()
 			result = {
 				"telephely": telephely,
 				"berelheto": bool(berelheto),
 				"rendszam": rendszam,
 				"alvazszam": alvazszam,
-				"uzembehelyezes": uzembehelyezes.isoformat(),
-				"muszaki_ervenyesseg": muszaki_ervenyesseg.isoformat(),
+				"uzembehelyezes": uzembehelyezes,
+				"muszaki_ervenyesseg": muszaki_ervenyesseg,
 				"marka": marka,
 				"tipus": tipus,
 				"szin": szin,
 				"teljesitmeny": teljesitmeny,
 				"ulesek": ulesek,
 				"legkondicionalo": bool(legkondicionalo),
-				"neve_napja": __get_nevnap(uzembehelyezes)
+				"neve_napja": nevnap
 			}
-			return jsonify(result)
+			return result
 		finally:
 			cur.close()
 	finally:
 		conn.close()
 
+@app.route('/jarmuvek/<jarmu_kulcs>.json')
+def show_vehicle(jarmu_kulcs):
+	resp = __get_vehicle(jarmu_kulcs)
+	if(resp == None):
+		abort(404)
+	return jsonify(resp)
+
+def __search(by, attr):
+	conn = get_db()
+	try:
+		cur = conn.cursor()
+		try:
+			cur.execute("select jarmu_kulcs from jarmu where %s like :attribute" % by, attribute=("%%"+attr+"%%")) # TODO: escape %
+			ids = []
+			for id in cur:
+				ids.append(id[0])
+			logging.debug("Searched by %s. The selected vehicles are: %s" % (by, str(ids)))
+			return ids
+		finally:
+			cur.close()
+	finally:
+		conn.close()
+
+@app.route('/jarmuvek/telephely-szerint/<telephely>.json')
+def search_by_depot(telephely):
+	resp = __search("telephely", telephely)
+	if(resp == None):
+		abort(404)
+#	else:
+	vehicles = []
+	for id in resp:
+		vehicles.append(__get_vehicle(id))
+	return jsonify(jarmuvek=vehicles)
+
+@app.route('/jarmuvek/marka-szerint/<marka>.json')
+def search_by_brand(marka):
+	resp = __search("marka", marka)
+	if(resp == None):
+		abort(404)
+#	else
+	vehicles = []
+	for id in resp:
+		vehicles.append(__get_vehicle(id))
+	return jsonify(jarmuvek=vehicles)
+
 def get_db():
 	"""Connects to the RDBMS and returns a connection object"""
 	# when used with a `file` object, `with` ensures it gets closed
@@ -202,7 +255,6 @@ def get_db():
 		config = json.load(config_file)
 	return cx_Oracle.connect(config['user'], config['pass'], config['host'])
 
-
 if __name__ == "__main__":
 	# pylint: disable=wrong-import-position,wrong-import-order
 	import os