def fill_form():

    form = UpdatePatient()
    ssnid = request.args['messages']
    room_amount = 0
    pat = Registration.objects(patient_ssnid=ssnid).first()

    form.patient_ssnid.data = ssnid
    if form.submit.data:
        if request.method == "POST":
            name = request.form["patientname"]
            age = request.form["age"]
            address = request.form["address"]
            #state = request.form["state"]
            #city = request.form["city"]
            state = str(dict(form.state.choices).get(form.state.data))
            city = str(dict(form.city.choices).get(form.city.data))
            bed_type = str(
                dict(form.type_of_bed.choices).get(form.type_of_bed.data))
            if bed_type == "General Ward":
                room_amount = 2000
            elif bed_type == "Sharing Ward":
                room_amount = 4000
            else:
                room_amount = 8000
            Registration.objects(patient_ssnid=ssnid).update(
                patient_name=name,
                bed_type=bed_type,
                age=age,
                address=address,
                state=state,
                city=city,
                room_amount=room_amount)
            flash("Patient Update initiated successfully", "success")
    return render_template("updatepatient.html", form=form, pat=pat)
def fill_form1():

    form = UpdatePatient()
    ssnid = request.args['msg']
    pat = Registration.objects(patient_ssnid=ssnid).first()
    form.patient_ssnid.data = ssnid
    bed_type = str(dict(form.type_of_bed.choices).get(form.type_of_bed.data))
    if form.delete.data:
        Registration.objects(patient_ssnid=ssnid).delete()
        GiveMedicine.objects(patient_ssnid=ssnid).delete()
        IssueDiagnostics.objects(patient_ssnid=ssnid).delete()
        flash("Patient Deletion Initiated Succefully", "success")
    return render_template("deletepatient.html", form=form, pat=pat)
def search_patient():

    form = UpdatePatient()
    if not session.get('login_id'):
        return redirect(url_for('admin_login'))
    patient_ssnid = form.patient_ssnid.data
    pat = Registration.objects(patient_ssnid=patient_ssnid).first()
    if form.get.data:
        if Registration.objects(patient_ssnid=patient_ssnid):
            flash("Patient Found", "success")
            return fill_form2(patient_ssnid)
        else:
            flash("This ID is not registered", "danger")

    return render_template("searchpatient.html", form=form, pat=pat)
def fill_form2(data):
	
	form= UpdatePatient()
	pat=Registration.objects(patient_ssnid=data).first()
	if form.submit.data:
		if request.method == "POST":
			name = request.form["patientname"]
			age = request.form["age"]
			address = request.form["address"]
			state = request.form["state"]
			city = request.form["city"]

			Registration.objects(patient_ssnid=data).update(patient_name=name,age=age,address=address,state=state,city=city)
			flash("Patient update initiated successfully","success")
	return render_template("searchpatient.html",form=form,pat = pat)
def delete_patient():

    form = UpdatePatient()
    if not session.get('login_id'):
        return redirect(url_for('admin_login'))
    patient_ssnid = form.patient_ssnid.data
    pat = Registration.objects(patient_ssnid=patient_ssnid).first()
    bed_type = str(dict(form.type_of_bed.choices).get(form.type_of_bed.data))
    if form.get.data:
        if Registration.objects(patient_ssnid=patient_ssnid):
            flash("Patient Found", "success")
            return redirect(url_for('fill_form1', msg=patient_ssnid))
        else:
            flash("This ID is not registered", "danger")

    return render_template("deletepatient.html", form=form, pat=pat)