payment.py

from flask import redirect, render_template, request, url_for, Blueprint, flash, session
from db import db, cursor
import mysql.connector
from auth import login_required
from datetime import datetime, time
from utils import requires_role
import pdb
import pdfkit


payment = Blueprint('payment', __name__)

@payment.route('/payment')
@login_required
@requires_role('admin')
def display():
	fetch_query = '''
		SELECT p.PaymentID, if(p.TotalPrice = 0, '', p.TotalPrice) as formatted_Price, p.mode 
		FROM payment p
	'''
	cursor.execute(fetch_query)
	db.commit()
	data = cursor.fetchall()

	Null_query = 'SELECT p.PaymentID FROM payment p JOIN bookingslot b on p.PaymentID = b.BSlotID WHERE b.TimeFrom is Null and b.TimeTo is Null'
	cursor.execute(Null_query)
	NullID = cursor.fetchone()

	return render_template('view/payment.html', data=data, NullID=NullID)

@payment.route('/payment/add', methods=['GET', 'POST'])
@login_required
def add_data():
	print('add data mai gaya')
	VehicleID = request.args.get('VehicleID')
	print('vehicleID ke niche')
	if request.method == 'POST':
		print('post method mai gaya')
		PaymentID = session.get('BSlotID')
		print('PaymentID', PaymentID)
		VehicleID = request.form.get('VehicleID')
		print('VehicleID', VehicleID)
		mode = request.form['mode']
		print('mode', mode)
		cursor.execute('SELECT SNo FROM vehicle WHERE VehicleID=%s', (VehicleID,))
		print('cursor.execute')
		db.commit()
		SNo = cursor.fetchone()
		print('Sno', SNo)
		if not SNo:
			flash('An error occurred. Please try again later.','error')
			return redirect(url_for('payment.add_data', VehicleID=VehicleID, PaymentID=PaymentID))

		S_No = SNo[0]
		if not S_No:
			flash('S_No nahi mil raha bhai','error')
			return redirect(url_for('payment.add_data', VehicleID=VehicleID, PaymentID=PaymentID))
		try:
			print('try mai gya fetch wale')
			fetchData = '''
				SELECT v.VehicleType, b.duration FROM vehicle v 
				JOIN bookingslot b ON v.SNo = b.SNo
				WHERE v.SNo=%s and v.VehicleID=%s
				'''
			cursor.execute(fetchData, (S_No, VehicleID,))
			v_data = cursor.fetchone()
			print('vdata', v_data)
			if not v_data:
				flash('An error occurred. Please try again later.','error')
				return redirect(url_for('payment.add_data', PaymentID=PaymentID, VehicleID=VehicleID))
		
			VehicleType = v_data[0]
			duration = v_data[1]
			Duration = int(duration)
			rate = 0

			if VehicleType == 'car':
				rate = 13

			elif VehicleType == '2-Wheeler':
				rate = 8

			elif VehicleType == 'Heavy-Vehicle':
				rate = 15

			elif VehicleType == 'Luxury-Vehicle':
				rate = 18

			TotalPrice = rate * Duration 
			TotalPrice = float(TotalPrice)
			session['TotalPrice'] = TotalPrice
			if TotalPrice == 0.0:
				flash('There was a server failure. Please book the slot agian', 'error')
				return redirect(url_for('vehicle.ChooseVehicle'))

		except mysql.connector.Error as e:
			print('except mai gya')
			db.rollback()
			print(e)
			flash('Error adding data', 'error')
			return redirect(url_for('payment.add_data',SNo=S_No, VehicleID=VehicleID, PaymentID=PaymentID))
		try:
			print('try muagya update wale')
			update_query = '''
				UPDATE payment
				SET TotalPrice=%s, mode=%s, SNo=%s, VehicleID=%s
				WHERE PaymentID=%s
			'''
			cursor.execute(update_query, (TotalPrice, mode, S_No, VehicleID, PaymentID,))
			db.commit()
		except mysql.connector.Error as e:
			print('except maigay')
			db.rollback()
			print(e)
			flash('Error adding your data','error')
			return redirect(url_for('payment.add_data', VehicleID=VehicleID, PaymentID=PaymentID, SNo=SNo))
	   
		try:
			print('sab kuch fetch wale try mau gaya')
			fetch_query = '''
				SELECT v.VehicleID, v.VehicleType, v.VehicleNumber, 
					   b.Date, b.TimeFrom, b.TimeTo, b.duration, 
					   u.SNo, u.username,     
					   o.name, o.contact,     
					   p.TotalPrice, p.mode 
				FROM vehicle v
				JOIN bookingslot b ON v.SNo = b.SNo
				JOIN user u ON v.SNo = u.SNo 
				JOIN owner o ON u.SNo = o.SNo 
				JOIN payment p ON b.BSlotID = p.PaymentID 
				WHERE v.SNo = %s  
				ORDER BY b.Date DESC, b.TimeTo DESC 
				LIMIT 1
					   
			'''
			cursor.execute(fetch_query, (SNo))
			data = cursor.fetchone()
			print('data', data)
			if data is None:
				print('data is None')
				flash('An error occurred. Please try again later', 'error')
				return redirect(url_for('payment.add_data', VehicleID=VehicleID, SNo=SNo))

			VehicleID = data[0]
			VehicleType = data[1]
			VehicleNumber = data[2]
			date = data[3]
			TimeFrom = data[4]
			TimeTo = data[5]
			duration = data[6]
			SNo = data[7]
			username = data[8]
			name = data[9]
			contact = data[10]
			TotalPrice = data[11]
			mode = data[12]

		except mysql.connector.Error as e:
			print('except mai gay' ,e)
			db.rollback()
			flash('Error processing your payment', 'error')
			return redirect(url_for('payment.add_data'))  

		try:
			print('insert_query wale try mai gya')
			insert_query = '''
				INSERT INTO allotment(VehicleID, SNo, username, date, TimeFrom, TimeTo, duration, name, contact, TotalPrice, mode, VehicleType, VehicleNumber) 
				VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)                '''
			# cursor.execute(insert_query, (VehicleID, username, date, TimeFrom, TimeTo, Duration, name, contact, TotalPrice, mode, VehicleType, VehicleNumber))
			cursor.execute(insert_query, (VehicleID, SNo, username, date, TimeFrom, TimeTo, Duration, name, contact, TotalPrice, mode, VehicleType, VehicleNumber))
			db.commit()

			# if(TotalPrice==0):
			#     db.rollback()
			#     flash('An error occurred. Please try again later','error')
			#     return redirect(url_for('index'))

		except mysql.connector.Error as e:
			print('except mai gays')
			print(e)
			flash('Error adding allotment', 'error')
			return redirect(url_for('payment.add_data'))

			# return render_template('add/payment.html',duration=duration, TotalPrice=TotalPrice, PaymentID=PaymentID)

		
		return redirect(url_for('payment.Generate_Receipt',VehicleID=VehicleID, duration=duration, TotalPrice=TotalPrice, mode=mode, PaymentID=PaymentID))

	
	Amount = session.get('TotalPrice')
	print('VehicleID: ', VehicleID)
	return render_template('add/payment.html', Amount=Amount, VehicleID=VehicleID)



def receipt(PaymentID):
	return render_template('view/GenerateReceipt.html',PaymentID=PaymentID)


@payment.route('/payment/edit/<int:PaymentID>', methods = ['GET','POST'])
@login_required
def edit_data(PaymentID):
	if request.method == 'POST':
		TotalPrice = request.form['TotalPrice']
		Mode = request.form['Mode']
		try:
			update_query = '''UPDATE payment
			SET TotalPrice=%s, Mode=%s
			WHERE PaymentID=%s'''
			cursor.execute(update_query, (TotalPrice, Mode, PaymentID,))
			#flash('Data updated successfully')
			return redirect(url_for('payment.display'))
		except mysql.connector.Error as e:
			db.rollback()
			flash('Error updating data')
	fetch_query = 'SELECT PaymentID, TotalPrice, Mode FROM payment WHERE PaymentID=%s'
	cursor.execute(fetch_query, (PaymentID,))
	db.commit()
	data = cursor.fetchone()
	if data is None:
		flash('No data found')
		return redirect(url_for('payment.display'))
	return render_template('edit/payment.html', data=data)

@payment.route('/payment/delete/<int:PaymentID>', methods = ['GET','POST'])
@login_required
def delete_data(PaymentID):
	if request.method == 'POST':

		try:
			delete_query = 'DELETE FROM payment WHERE PaymentID = %s'
			cursor.execute(delete_query, (PaymentID,))
			db.commit()
			#flash('Data deleted successfully')
			return redirect(url_for('payment.display'))
		except mysql.connector.Error as e:
			db.rollback()
			flash('Error deleting data')
	fetch_query = 'SELECT PaymentID, TotalPrice, Mode FROM payment WHERE PaymentID = %s'
	cursor.execute(fetch_query, (PaymentID,))
	db.commit()
	data = cursor.fetchone()
	if data is None:
		flash('No data found')
		return redirect(url_for('payment.display'))
	return render_template('delete/payment.html',data=data)

@payment.route('/payment/generate_receipt/<int:PaymentID>', methods=['GET', 'POST'])
@login_required
def Generate_Receipt(PaymentID):
	VehicleID = request.args.get('VehicleID')
	print('VehicleID: ', VehicleID)
	try:
		fetch_query = '''
			SELECT v.SNo, v.VehicleType, v.VehicleNumber, b.date, p.PaymentID, p.TotalPrice, p.Mode, b.TimeFrom, b.TimeTo, v.VehicleID
			FROM payment p 
			JOIN bookingslot b ON p.PaymentID = b.BSlotID 
			JOIN vehicle v  ON p.SNo = v.SNo
			WHERE p.PaymentID=%s and v.VehicleID=%s
		'''
		cursor.execute(fetch_query, (PaymentID,VehicleID,))
		db.commit()
		data = cursor.fetchone()
		print(data)
		print(session['role'])
		if data is None:
			flash('Slot not booked. Please book the slot to generate receipt', 'danger')
			return redirect(url_for('payment.display'))
		if session['role'] == 'user':
			if data[5] == 0.0:
				return redirect(url_for('vehicle.add_data',SNo=data[0]))
		elif session['role'] == 'admin':
			if data[5] == 0.0:
				flash('An error occured. Please try again', 'error')
				return redirect(url_for('vehicle.AdminVehicle'))
		else:
			return "Server failed"


		SNo = data[0]
		VehicleType = data[1]
		VehicleNumber = data[2]
		Date = data[3]
		ReceiptID = data[4]
		Price = data[5]
		Mode = data[6]
		TimeFrom = data[7]
		TimeTo = data[8]
		if Price == 0.0:
			flash('There was a server failure. Please book the slot agian', 'error')
			return redirect(url_for('vehicle.ChooseVehicle'))

		if TimeFrom is None or TimeTo is None:
			flash('Booking slot time data is missing', 'error')
			return redirect(url_for('payment.display'))

		duration = (datetime.strptime(str(TimeTo), '%H:%M:%S') - datetime.strptime(str(TimeFrom), '%H:%M:%S')).seconds / 3600

		rate=0
		if VehicleType == '2-Wheeler':
			rate = 8
		elif VehicleType == 'car':
			rate = 13
		elif VehicleType == 'Heavy-Vehicle':
			rate = 15
		elif VehicleType == 'Luxury-Vehicle':
			rate = 18
		Price = rate * duration
		
	except mysql.connector.Error as e:
		print(e)
		db.rollback()
		flash('Error generating receipt', 'error')
		return redirect(url_for('payment.display'))
	

	try:
		username = session.get('username')
		print(username)
		cursor.execute('SELECT * FROM allotment WHERE username=%s', (username,))
		db.commit()
	except mysql.connector.Error as e:
		print(e)
		db.rollback()
		return redirect(url_for('payment.Generate_Receipt', PaymentID=PaymentID))

	return render_template('view/GenerateReceipt.html', strftime=datetime.strftime, PaymentID=PaymentID,
							   VehicleType=VehicleType, VehicleNumber=VehicleNumber, date=Date, ReceiptID=ReceiptID, Price=Price, Mode=Mode, TimeFrom=TimeFrom, TimeTo=TimeTo, SNo=SNo)