From e5d413994cb23564abc9c42bf0ce2cc762222a6c Mon Sep 17 00:00:00 2001 From: Navan Chauhan Date: Sun, 26 May 2019 18:17:49 +0530 Subject: Created python folder --- pythonProgram/main.py | 342 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 342 insertions(+) create mode 100755 pythonProgram/main.py (limited to 'pythonProgram/main.py') diff --git a/pythonProgram/main.py b/pythonProgram/main.py new file mode 100755 index 0000000..2d89586 --- /dev/null +++ b/pythonProgram/main.py @@ -0,0 +1,342 @@ +"""AutoSafe +Modular Road-Safety Toolkit + +""" +#pylint: disable=C0103,C0111 +import argparse +import time +import overpy +import simplejson as sjson +from scipy.spatial import distance as dist +from imutils.video import VideoStream +from imutils import face_utils +#import playsound +import imutils +import dlib +import cv2 +#from uber_rides.session import Session +#from uber_rides.client import UberRidesClient +import tweepy +import requests + +#import geocoder +#import obd + +###################### +# Defining Variables # +###################### + +radius = str(100) # Radius for maxspeed function. taken as a string because gets concatenated + +################ +# Dummy Values # +################ + + +#Latitude +glat = 28.544565 #28.546519 +#Longitude +glng = 77.193320 #77.179248 + +# Commented this as we are not on a road I guess? + +#g = geocoder.ip('me') +#print(g.latlng) + +##################### +# Real Time Values # +##################### + +#glat = g.lat +#glng = g.lng + +################### +# Dummy OBD Setup # +################### + +############################################################## +# What the commands are doing: # +# It setups and asynchronous watch over the speed of the car # +# this means that the speed's data is constantly updated # +# This makes the while loop work # +############################################################## + + +#connection = obd.Async() # auto-connects to USB or RF port +#connection.watch(obd.commands.SPEED) # select an OBD command (sensor) +#connections.start() +#carSpeed = connection.query(obd.commands.SPEED) # send the command, and parse the response + +#carSpeed = 30 + +##################### +# Tweeting Function # +##################### + +def get_api(cfg): + auth = tweepy.OAuthHandler(cfg['consumer_key'], cfg['consumer_secret']) + auth.set_access_token(cfg['access_token'], cfg['access_token_secret']) + return tweepy.API(auth) + +def tweetMe(): + cfg = { + "consumer_key":"knQFpTnjuSvr6OxYwebt3wyrd", + "consumer_secret":"Mhex3oRkmaF7lD3hoMvHpAD6ctW0ugKYCopTlhc0JzOLOMIZ0w", + "access_token":"2846631344-wEozinvHfEIFxFVy51I6te8SrN5OTFtU00wxsiz", + "access_token_secret":"Nfx1U8a2TjAQXFLBrJIyy2p36sjBGAWFIthLc1cIoI56U" + } + + api = get_api(cfg) + headers = { + 'Accept': 'application/json', + 'user-key': 'a530c1424d9abe5442fa22f77ce03d25', + } + + params = ( + ('lat', '28.546519'), + ('lon', '77.179248'), + ) + url = 'https://developers.zomato.com/api/v2.1/geocode' + response = requests.get(url, headers=headers, params=params) + loc = response.json()['location']['title'] + tweet = ("Stay Alert! Sudden braking at: ", loc) + status = api.update_status(status=tweet) + print(status) + +################################ +# Fetching Details from Zomato # +################################ + +def getRes(): + res = [] + headers = { + 'Accept': 'application/json', + 'user-key': 'a530c1424d9abe5442fa22f77ce03d25', + } + + params = ( + ('lat', '28.546519'), + ('lon', '77.179248'), + ) + url = 'https://developers.zomato.com/api/v2.1/geocode' + response = requests.get(url, headers=headers, params=params) + res = response.json()['popularity']['nearby_res'] + return res + +def getDetails(res): + headers = { + 'Accept': 'application/json', + 'user-key': 'a530c1424d9abe5442fa22f77ce03d25', + } + url = "https://developers.zomato.com/api/v2.1/restaurant?res_id=" + str(res[0]) + newResponse = requests.get(url, headers=headers) + #newRes = [] + resName = newResponse.json()['name'] + resAddress = newResponse.json()['location']['address'] + print("You are feeling sleepy, why don't you take a break?\n") + print("Your nearest eatery is " + resName, "\n") + print(resName + " is at " + resAddress, "\n") + +def zomato(): + res = getRes() + getDetails(res) + +############################ +# Combatting Drunk Driving # +############################ + +def drunk(): + print("Your appear to be around places which sell \n alcohol, take the breathalyser test") + + bac_raw = str(open("./files/bac.txt", "r").read()) + bac = float(float(bac_raw)/100) + print(bac) + if bac >= 0.08: + print("Please Do Not Drive!\n") + print("I can call a cab if you want\n") + print("") + answer = input("Say No, to disagree, else I'll book the cab: \n") + if answer == "no": + print("You are not fit to drive") + print("Text message to emergency contact sent") + print("Your cab has been booked, thank you for not driving") + print("have a safe journey!") + + +def sound_alarm(): + print('You Sleep You Lose') + +def eye_aspect_ratio(eye): + + # Computes the euclidean distances between the two sets of eyes + A = dist.euclidean(eye[1], eye[5]) + B = dist.euclidean(eye[2], eye[4]) + C = dist.euclidean(eye[0], eye[3]) + + ear = (A + B) / (2.0 * C) + + return ear + +shape_predictor = "./files/shape_predictor_68_face_landmarks.dat" + + +EYE_AR_THRESH = 0.2 # If the EAR goes < this for 48 frames, it is counted as drowsiness +EYE_AR_CONSEC_FRAMES = 24 + + + + +######################## +# Drowsiness Detection # +######################## + +def sleepiness(): + + COUNTER = 0 + #ALERT = False + + print("Initialising Facial Landmark Predictor...") + detector = dlib.get_frontal_face_detector() + predictor = dlib.shape_predictor(shape_predictor) + + (lStart, lEnd) = face_utils.FACIAL_LANDMARKS_IDXS["left_eye"] + (rStart, rEnd) = face_utils.FACIAL_LANDMARKS_IDXS["right_eye"] + + print("Starting Video Stream...") + vs = VideoStream(src=0).start() + time.sleep(1.0) + while True: + frame = vs.read() + frame = imutils.resize(frame, width=450) + gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) + + rects = detector(gray, 0) + for rect in rects: + shape = predictor(gray, rect) + shape = face_utils.shape_to_np(shape) + + leftEye = shape[lStart:lEnd] + rightEye = shape[rStart:rEnd] + leftEAR = eye_aspect_ratio(leftEye) + rightEAR = eye_aspect_ratio(rightEye) + + ear = (leftEAR + rightEAR) / 2.0 + + leftEyeHull = cv2.convexHull(leftEye) + rightEyeHull = cv2.convexHull(rightEye) + cv2.drawContours(frame, [leftEyeHull], -1, (0, 255, 0), 1) + cv2.drawContours(frame, [rightEyeHull], -1, (0, 255, 0), 1) + + if ear < EYE_AR_THRESH: + COUNTER += 1 + + # if the eyes were closed for a sufficient number of + # then sound the alarm + if COUNTER >= EYE_AR_CONSEC_FRAMES: + # if the alarm is not on, turn it on + if not ALARM_ON: + ALARM_ON = True + + # draw an alarm on the frame + cv2.putText(frame, "Sleepiness Detected!", (10, 30), + cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255), 2) + print("Sleepiness Detected!") + print("\n \n \n") + zomato() + time.sleep(5) + exit() + + + + else: + COUNTER = 0 + ALARM_ON = False + + cv2.putText(frame, "Ratio: {:.2f}".format(ear), (300, 30), + cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255), 2) + + cv2.imshow("Frame", frame) + key = cv2.waitKey(1) & 0xFF + + # breaks loop on q + if key == ord("q"): + break + + cv2.destroyAllWindows() + vs.stop() + +############################## +# Speed Limit Fetch Function # +############################## + +def maxspeed(coordinates, radius): + + lat, lon = coordinates + api = overpy.Overpass() + + ####################### + # Query for Open Maps # + ####################### + + result = api.query(""" + way(around:""" + radius + """,""" + lat + """,""" + lon + """) ["maxspeed"]; + (._;>;); + out body; + """) + + results_list = [] + for way in result.ways: + road = {} + road["name"] = way.tags.get("name", "n/a") + road["speed_limit"] = way.tags.get("maxspeed", "n/a") + nodes = [] + for node in way.nodes: + nodes.append((node.lat, node.lon)) + road["nodes"] = nodes + results_list.append(road) + return results_list + + + + +######################################################################### +# Gives data to the function and gets json in return. # +#This json is then parsed. Then the double quotes are stripped off it # +######################################################################### +def speedlim(): + speedLimit = sjson.dumps(maxspeed((str(glat), str(glng)), radius)[0]['speed_limit']).strip('\"') + while True: + carSpeedDummy = open("./files/carSpeed.txt", "r") + carSpeed = carSpeedDummy.read() + #print(carSpeed) + #carSpeed = 29 + while int(carSpeed) > int(speedLimit): + carSpeedDummy = open("./files/carSpeed.txt", "r") + carSpeed = carSpeedDummy.read() + print("Over The Speed Limit") + time.sleep(5) + while int(carSpeed) <= int(speedLimit): + carSpeedDummy = open("./files/carSpeed.txt", "r") + carSpeed = carSpeedDummy.read() + print("Under the Speed Limit") + time.sleep(5) + +################## +# Sudden Braking # +################## + +def brakes(): + while True: + tweetMe() + +##################################################### +# Using Argument Parse to run one command at a time # +##################################################### + +parser = argparse.ArgumentParser() +FUNCTION_MAP = {'overspeed' : speedlim, 'sleep-detector' : sleepiness, + 'sudden-braking' : brakes, 'drunk' : drunk} +parser.add_argument('function', choices=list(FUNCTION_MAP)) +args = parser.parse_args() +FUNC = FUNCTION_MAP[args.function] +FUNC() -- cgit v1.2.3