app.py

'''
        Ride-a-Bike
        Developer : Rakshit Deshpande
'''
import pymongo,datetime,hashlib,math,pytz,random
from flask import Flask, render_template, url_for, redirect, request ,flash ,session
from pymongo import MongoClient
from flask_mail import Mail,Message
from fpdf import FPDF
from flask import send_file
import os

app = Flask(__name__)

app.secret_key = 'Ride-a-Bike'

#accesing environment variables
manager_name = os.environ['MANAGER_NAME']
manager_pass = os.environ['MANAGER_PASS']
db_username = os.environ['DB_USERNAME']
db_password = os.environ['DB_PASS']
id = os.environ['MAIL_PASS']
mail_id = os.environ['MAIL_ID']

client = pymongo.MongoClient("mongodb+srv://"+db_username+":"+db_password+"@cluster0-2ogac.mongodb.net/test?retryWrites=true&w=majority")
db = client["ride-a-bike"]

app.config['MAIL_SERVER']='smtp.gmail.com'
app.config['MAIL_PORT'] = 465
app.config['MAIL_USERNAME'] = mail_id
app.config['MAIL_PASSWORD'] = id
app.config['MAIL_USE_TLS'] = False
app.config['MAIL_USE_SSL'] = True
mail = Mail(app)

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/signup',methods = ['POST','GET'])
def signup():
    if request.method == 'GET':
        return render_template('signup.html')
    else:
        if request.form['name']=="" or request.form['email']=="" or request.form['phone_number']=="" or request.form['gender']=="" or request.form['blood_group']=="" or request.form['dob']=="" or request.form['dl_number']=="" or request.form['dl_valid_till']=="" or request.form['password']=="" :
            return redirect('/signup')
        password = request.form['password']
        pass_256 = hashlib.sha256(password.encode())
        pass_encrypt = pass_256.hexdigest()
        cred = {"name":request.form['name'],"email":request.form['email'],"phone_number":request.form['phone_number'],"gender":request.form['gender'],"blood_group":request.form['blood_group'],"dob":request.form['dob'],"dl_number":request.form['dl_number'],"dl_valid_till":request.form['dl_valid_till'],"password":pass_encrypt,"balance":0,"status":"-"}
        db.details.insert(cred) 
        session['username'] = request.form['name']
        return redirect('/home')

@app.route('/login', methods=['POST','GET'])
def login():
    if request.method=="GET":
        return render_template("login.html")
    else:
        try:
            # if request.form['name'] == admin_name and request.form['password'] == admin_pass:
            #     session['username'] = request.form['name']
            #     return "admin"
            if request.form['name'] == manager_name and request.form['password'] == manager_pass:
                session['username'] = request.form['name']
                return redirect('/manager')
            else:
                password = request.form['password']
                pass_256 = hashlib.sha256(password.encode())
                pass_encrypt = pass_256.hexdigest()
                x = db.details.find({"name":request.form['name']})
                if x[0]["password"] == pass_encrypt :
                    session['username'] = request.form['name']
                    if x[0]["status"] == "riding" :
                        return redirect('/end_ride')
                    return redirect('/home')
                else:
                    return redirect('/login')
        except:
            error = "Invalid User Credentials"
            return render_template("login.html",error = error) 

@app.route('/home',methods = ['POST','GET'])
def home():
    if 'username' in session and session['username'] != manager_name:
      if request.method == 'GET':
        username = session['username']
        scooter_details = db.scooter.find({})
        docking_station_details = db.docking_station.find({})
        amt = db.details.find({"name":session['username']})
        amount = amt[0]["balance"]
        x = db.fare.find({})
        return render_template("home.html",username = username,docking_station_details=docking_station_details,amount = amount,base_price = x[0]["base_price"],per_min = x[0]["per_min"],min_price = x[0]["min_price"])
      else:
          
          return redirect('/start_ride')
    else:
        return render_template("login_error.html")

@app.route('/manager')
def manager():
    try:
        if 'username' in session and session['username'] == manager_name:
            data = db.details.find({"status":"riding"})
            logs = db.logs.find({}).sort([("start_time",pymongo.DESCENDING)])
            return render_template("manager.html", data = data, logs = logs)
        return render_template("login_error.html")
    except:
        return redirect('/manager')



@app.route('/add_scooter', methods = ['POST','GET'])
def add_scooter():
    try:
        if 'username' in session and session['username'] == manager_name:
            if request.method == 'GET' :
                docking_station_details = db.docking_station.find({})
                x = db.details.find({"name":session['username']})
                amount = x[0]["balance"]
                return render_template("add_scooter.html",docking_station_details = docking_station_details,amount = amount)
            else:
                data = {"registration_number":request.form['registration_number'],"insurance_number":request.form['insurance_number'],"insurance_valid_till":request.form['insurance_valid_till'],"docking_station":request.form["docking_station"],"ignition_status":"off","rider_name":"-"}
                x = db.scooter.find({"registration_number":request.form['registration_number']})
                y = db.scooter.find({"insurance_number":request.form['insurance_number']})
                docking_station_details = db.docking_station.find({})
                try :
                    if x[0] :
                        error = "A vehicle with registration number already exists!"
                        x = db.details.find({"name":session['username']})
                        amount = x[0]["balance"]
                    return render_template("add_scooter.html",error = error,docking_station_details = docking_station_details,amount = amount)
                except:
                    try:
                        if y[0] :
                            error = "Insurance Number is of different vehicle"
                            x = db.details.find({"name":session['username']})
                            amount = x[0]["balance"]
                            return render_template("add_scooter.html",error = error,docking_station_details = docking_station_details,amount = amount)
                    except:
                        db.scooter.insert(data)
                        x = db.docking_station.find({"station_name":request.form['docking_station']})
                        num = x[0]["no_of_scooters"]
                        num = num + 1
                        db.docking_station.update({"station_name":request.form['docking_station']},{"$set":{"no_of_scooters":num}})
                        return redirect('/manager')
        else:
            return render_template("login_error.html")
    except:
        return render_template("add_scooter.html")

@app.route('/start_ride', methods = ['POST','GET'])
def start_ride():
    try:
        if 'username' in session and session['username'] != manager_name:
            if request.method == 'POST': 
                username = session['username']
                scooter_data = db.scooter.find({"docking_station":request.form['station_name']})
                db.details.update({"name":session['username']},{"$set":{"from":request.form['station_name']}})
                x = db.details.find({"name":session['username']})
                amount = x[0]["balance"]
                y = db.fare.find({})
                return render_template("start_ride.html",username = username,scooter_data = scooter_data,amount = amount,base_price = y[0]["base_price"],per_min = y[0]["per_min"],min_price = y[0]["min_price"])
            else:
                x = db.details.find({"name":session['username']})
                amount = x[0]["balance"] 
                y = db.fare.find({})
                return render_template("start_ride.html",amount = amount,username = session['username'],base_price = y[0]["base_price"],per_min = y[0]["per_min"],min_price = y[0]["min_price"])
        else:
            return render_template("login_error.html")
    except:
        return redirect('/start_ride')

@app.route('/end_ride', methods = ['POST','GET'])
def end_ride():
    try:
        if 'username' in session and session['username'] != manager_name:
            if request.method == 'POST':
                x = db.details.find({"name":session['username']})
                # if x[0]["status"] == "-" :
                #     return redirect('/home')
                bal = db.details.find({"name":session['username']})
                balance = bal[0]["balance"]
                x = db.fare.find({})
                min_price = int(x[0]["min_price"])
                if balance < min_price:
                    error = "No sufficcient balance, please Top Up your wallet"
                    return render_template("start_ride.html",balance = balance,username = session['username'],error = error)
                username = session['username']
                docking_station_data = db.docking_station.find({})
                a = datetime.datetime.now(pytz.timezone('Asia/Calcutta'))
                start_time = a.strftime("%c")
                data = db.scooter.find({"registration_number":request.form['registration_number']})
                station  = data[0]["docking_station"]
                db.details.update({"name":session['username']},{"$set":{"registration_number":request.form['registration_number'],"start_time":start_time,"status":"riding"}})
                db.scooter.update({"registration_number":request.form['registration_number']},{"$set":{"rider_name":session['username'],"ignition_status":"on","docking_station":"-"}})
                details = db.docking_station.find({"station_name":station})
                num = details[0]["no_of_scooters"]
                num = num - 1
                db.docking_station.update({"station_name":station},{"$set":{"no_of_scooters":num}})
                return render_template("end_ride.html",username = username,docking_station_data = docking_station_data)
            else:
                x = db.details.find({"name":session['username']})
                if x[0]["status"] == "-" :
                    return redirect('/home')
                username = session['username']
                docking_station_data = db.docking_station.find({})
                return render_template("end_ride.html",username = username,docking_station_data = docking_station_data)
        else:
            return render_template("login_error.html")
    except:
        return render_template("start_ride.html")

    
@app.route('/bill',methods = ['POST','GET'])
def bill():
    try:
        if 'username' in session and session['username'] != manager_name:
          if request.method == 'POST':
            a = datetime.datetime.now(pytz.timezone('Asia/Calcutta'))
            end_time = a.strftime("%c")
            db.details.update({"name":session['username']},{"$set":{"destination":request.form['destination'],"end_time":end_time}})
            db.scooter.update({"rider_name":session['username']},{"$set":{"docking_station":request.form['destination'],"ignition_status":"off","rider_name":"-"}})
            details = db.docking_station.find({"station_name":request.form['destination']})
            num = details[0]["no_of_scooters"]
            num = num + 1
            db.docking_station.update({"station_name":request.form['destination']},{"$set":{"no_of_scooters":num}})
            data = db.details.find({"name":session['username']})
            start_time = data[0]["start_time"]
            a = start_time.split(" ")
            x = db.fare.find({})
            base_price =  int(x[0]["base_price"])
            per_min = int(x[0]["per_min"])
            min_price = int(x[0]["min_price"])
            global time
            global before_hour
            global before_min
            try :
                time = a[3].split(":")
                before_hour = int(time[0]) 
                before_min = int(time[1])
            except:
                time = a[4].split(":") 
                before_hour = int(time[0])
                before_min = int(time[1])
            a = end_time.split(" ")
            global after_hour
            global after_min
            try :
                time = a[3].split(":")
                after_hour = int(time[0])
                after_min = int(time[1])
            except:
                time = a[4].split(":")
                after_hour = int(time[0])
                after_min = int(time[1])

            #calculaitng ride timming
            hour = after_hour - before_hour -1
            minutes = after_min + (60 - before_min) + (hour*60)
            #caluclation fare on half an hour basis
            # quo = int(minutes/30)
            # rem = minutes%30
            # if rem == 0:
            #     amount = quo*50
            # else:
            #     amount = (quo + 1 )*50
            amount = base_price + (minutes * per_min)
            balance = data[0]["balance"]
            balance = balance - amount
            db.details.update({"name":session['username']},{"$set":{"balance":balance}})
            details = {"name":session['username'],"registration_number":data[0]["registration_number"],"from":data[0]["from"],"start_time":data[0]["start_time"],"destination":data[0]["destination"],"end_time":data[0]["end_time"],"amount":amount,"duration":minutes}
            db.logs.insert(details)

            #writing into pdf file
            pdf = FPDF()
            pdf.add_page()
            pdf.set_font("times",'I', size=32)
            pdf.cell(200, 10, txt="Ride-a-Bike", ln=1, align="C")
            pdf.set_font("Arial", size=20)
            pdf.set_line_width(1)
            pdf.set_draw_color(255, 0, 0)
            pdf.line(10, 25, 200, 25)
            pdf.image('static/images/logo.png', x=85, y=30, w=50)
            pdf.set_font("Arial",'B', size = 20)
            pdf.cell(200,120,txt = session['username'] , ln = 1, align="L")
            pdf.set_font("Arial", size = 18)
            pdf.set_line_width(1)
            pdf.set_draw_color(0, 0, 0)
            pdf.line(15, 90, 195, 90)
            start_station = data[0]["from"]
            pdf.cell(200, -75, txt="From :  "+start_station+"    "+start_time , ln=1, align="L")
            pdf.cell(200,100, txt="To :   "+request.form['destination']+"    "+end_time , ln=1, align="L")
            pdf.cell(200,-75, txt = "Minimum Fare : "+str(base_price) ,ln = 1, align="L")
            pdf.cell(200,100, txt = "Time Fare ("+str(minutes)+" mins) : "+str((minutes * per_min)) ,ln = 1, align="L")
            pdf.set_font("Arial",'B', size = 20)
            pdf.cell(200,-75, txt = "Total : "+str(amount) ,ln = 1, align="L")
            pdf.set_line_width(1)
            pdf.set_draw_color(0, 0, 0)
            pdf.line(15, 165, 195, 165)
            pdf.output("ride-a-bike_bill.pdf")

            db.details.update({"name":session['username']},{"$set":{"status":"-"}})
            return render_template("bill.html",time = end_time,data = data,minutes = minutes,amount = amount,username = session['username'],balance = balance,base_price = x[0]["base_price"],per_min = x[0]["per_min"],min_price = x[0]["min_price"])
          else:
              return render_template("bill.html")
        else:
            return render_template("login_error.html")
    except:
        return redirect('/end_ride')

@app.route('/add_station',methods = ['POST','GET'])
def add_station():
    try :
      if 'username' in session and session['username'] == manager_name:
        if request.method == 'POST':
            data = {"station_name":request.form["station_name"],"no_of_scooters":0}
            db.docking_station.insert(data)
            return render_template("manager.html")
        else:
            return render_template("manager.html")
      else:
          render_template("login_error.html")
    except:
        return render_template("manager.html")

@app.route('/remove_scooter',methods = ['POST','GET'])
def remove_scooter():
    try :
        if 'username' in session and session['username'] == manager_name:
            if request.method == 'POST':
                return "request accepted!"
            else:
                return render_template("manager.html")
    except:
        return render_template("manager.html")
    
@app.route('/remove_station',methods = ['POST','GET'])
def remove_station():
    try :
        if 'username' in session and session['username'] == manager_name:
            if request.method == 'POST':
                return "request accepted!"
            else:
                return render_template("manager.html")
    except:
        return render_template("manager.html")

@app.route('/rides')
def rides():
    try :
        if 'username' in session and session['username'] != manager_name:
            data = db.logs.find({"name":session['username']}).sort([("start_time",pymongo.DESCENDING)])
            x =  db.details.find({"name":session['username']})
            amount = x[0]["balance"]
            y = db.fare.find({})
            return render_template("rides.html", data = data ,username = session['username'],amount = amount,base_price = y[0]["base_price"],per_min = y[0]["per_min"],min_price = y[0]["min_price"])
        else:
            return render_template("login_error.html")
    except:
        return redirect('/home')

@app.route('/account_settings',methods = ['POST','GET'])
def accout_settings():
    try:
        if 'username' in session and session['username'] != manager_name:
            if request.method == 'POST':
                if request.form['email'] != "":
                    db.details.update({"name":session['username']},{"$set":{"email":request.form['email']}})
                if request.form['phone_number'] != "":
                    db.details.update({"name":session['username']},{"$set":{"phone_number":request.form['phone_number']}})
                if request.form['dl_valid_till'] != "":
                    db.details.update({"name":session['username']},{"$set":{"dl_valid_till":request.form['dl_valid_till']}})
                if request.form['password'] != "":
                    password = request.form['password']
                    pass_256 = hashlib.sha256(password.encode())
                    pass_encrypt = pass_256.hexdigest()
                    db.details.update({"name":session['username']},{"$set":{"password":pass_encrypt}})
                return redirect('/home')
            else:
                x = db.fare.find({})
                return render_template("update.html", username = session['username'],base_price = x[0]["base_price"],per_min = x[0]["per_min"],min_price = x[0]["min_price"])
        else:
            return render_template("login_error.html")
    except:
        return redirect('/home')

@app.route('/top_up',methods = ['POST','GET'])
def top_up():
    try:
        if 'username' in session and session['username'] != manager_name:
                if request.method == 'POST':
                    amt = request.form['Amount']
                    x = db.details.find({"name":session['username']})
                    a = x[0]["balance"]
                    amount = int(a) + int(amt)
                    db.details.update({"name":session['username']},{"$set":{"balance":amount}})
                    a = datetime.datetime.now(pytz.timezone('Asia/Calcutta'))
                    time = a.strftime("%c")
                    data = {"name":session['username'],"time":time,"amount":int(amt),"mode":"debited"}
                    db.payments.insert(data)
                    return redirect('/home')
    except:
        return render_template("login_error.html")

@app.route('/forgot_pass',methods = ['POST','GET'])
def forgot_pass():
    if request.method == 'GET':
        return redirect('/home')
    else:
        Name = request.form['name']
        print(Name)
        file = open("name",'w')
        file.write(Name)
        file.close()
        return render_template("verify.html")

@app.route('/verify_code',methods = ['POST','GET'])
def verify_code():
    if request.method == 'GET':
        try :
            global code
            code = random.randrange(100000,999999)
            file = open("name","r")
            name = file.read()
            x = db.details.find({"name":name})
            email = x[0]["email"]
            msg = Message('OTP', sender = mail_id, recipients = [email])
            msg.body = "Yours OTP is "+str(code)+" please do not share this OTP with someone"
            mail.send(msg)
            print("Send mail ",code," to ",x[0]["email"])
            return render_template("verify.html")
        except:
            error = "Invalid username"
            return render_template("login.html",error = error)
    else:
        Code = request.form['code']
        if code == int(Code):
            file = open("name","r")
            name = file.read()
            session['username'] = name
            print(name)
            message = "Update your password"
            return render_template('update.html',message = message,username = session['username'])
        else:
            error = "Incorrect OTP!!"
            return render_template("login.html",error = error)
        
@app.route('/payments')
def payments():
    try:
        if 'username' in session and session['username'] != manager_name:
            data = db.payments.find({"name":session['username']}).sort([("time",pymongo.DESCENDING)])
            a = db.details.find({"name":session['username']})
            amount = a[0]["balance"]
            x = db.fare.find({})
            return render_template("payments.html",data = data,username = session['username'],amount = amount,base_price = x[0]["base_price"],per_min = x[0]["per_min"],min_price = x[0]["min_price"]) 
    except:
        return render_template("login_error.html")

@app.route('/download_bill')
def download_bill():
    if 'username' in session and session['username'] != manager_name:
        return send_file("ride-a-bike_bill.pdf", as_attachment=True)
    else:
        return render_template("login_error.html")

@app.route('/my_account')
def my_account():
    if 'username' in session and session['username'] != manager_name:
        username = session['username']
        x = db.details.find({"name":username})
        amount = x[0]["balance"]
        y = db.fare.find({})
        return render_template("my_account.html",username = username,amount = amount,details = x,base_price = y[0]["base_price"],per_min = y[0]["per_min"],min_price = y[0]["min_price"])
    else:
        return render_template("login_error.html")

@app.route('/delete_account')
def delete_account(methods = ['POST','GET']):
    if 'username' in session and session['username'] != manager_name:
        if request.method == 'POST':
            return redirect('/my_account')
        else:
            x = db.details.find({"name":session['username']})
            amount = int(x[0]["balance"])
            if amount < 0:
                msg = "Payment is due"
                y = db.fare.find({})
                return render_template("my_account.html",msg = msg,username = session['username'],amount = x[0]["balance"],details = x,base_price = y[0]["base_price"],per_min = y[0]["per_min"],min_price = y[0]["min_price"])
            else:
                try:
                    db.details.delete_one({"name":session['username']})
                    db.logs.delete_many({"name":session['username']})
                    db.payments.delete_many({"name":session['username']})
                    session.pop('username',None)
                    return redirect('/')
                except:
                    return redirect('/')
    else:
        return render_template("login_error.html")

@app.route('/messages',methods = ['POST','GET'])
def messages():
    if request.method == 'POST':
        name = request.form["name"]
        email = request.form["email"]
        message = request.form["message"]
        if name != "" and email != "" and message != "":
            db.messages.insert({"name":name,"email":email,"message":message})
        return redirect('/')
    else:
        if 'username' in session and session['username'] == manager_name:
            x = db.messages.find({})
            return render_template("view_messages.html",messages = x)
        else:
            return render_template("login_error.html")

@app.route('/set_fare' , methods = ['POST','GET'])
def set_fare():
    if 'username' in session and session['username'] == manager_name:
            if request.method == 'POST':
                base_price = request.form['base_price']
                per_min = request.form['per_min']
                min_price = request.form['min_price']
                db.fare.delete_many({})
                db.fare.insert({"base_price":base_price,"per_min":per_min,"min_price":min_price})
                # db.fare.update({"_id":*},{"$set":{"base_price":base_price,"per_min":per_min,"min_price":min_price}})
                return redirect('/manager')
            else:
                return render_template("set_fare.html")
    else:
        return render_template("login_error.html")

@app.route('/logout')
def logout():
    session.pop('username',None)
    return redirect('/')


@app.errorhandler(404)
def notFound(e):
    return render_template("404.html"), 404

@app.route('/delete_scooter')
def delete_scooter():
  try:
    if 'username' in session and session['username'] == manager_name:
        db.scooter.delete_many({})
        return render_template("index.html")
    else:
        render_template("login_error.html")
  except:
      return redirect("/")

@app.route('/delete_station')
def delete_station():
  try:
    if 'username' in session and session['username'] == manager_name:
        db.docking_station.delete_many({})
        return render_template("index.html")
    else:
        render_template("login_error.html")
  except:
      return redirect("/")

@app.route('/delete_details')
def delete_details():
  try:
    if 'username' in session and session['username'] == manager_name:
        db.details.delete_many({})
        return render_template("index.html")
    else:
        render_template("login_error.html")
  except:
      return redirect("/")

@app.route('/clear_logs')
def clear_logs():
  try:
    if 'username' in session and session['username'] == manager_name:
        db.logs.delete_many({})
        return redirect('/manager')
    else:
        return render_template("login_error.html")
  except:
      return redirect("/")
      
@app.route('/clear_payments')
def clear_payments():
  try:
    if 'username' in session and session['username'] == manager_name:
        db.payments.delete_many({})
        return redirect('/manager')
    else:
        return render_template("login_error.html")
  except:
      return redirect("/")
    
if __name__ == '__main__':
    app.run(host="0.0.0.0",port=5000, threaded = True, debug = True)