Wednesday, August 14, 2024

Complete Railway Reservation System Using SQLITE & STREAMLIT

                                     INTRODUCTION 

Streamlit is a free and open-source framework to rapidly build and share beautiful machine learning and data science web apps. It is a Python-based library specifically designed for machine learning engineers.


DOWNLOAD LINK 


Watch On Youtube

https://www.youtube.com/watch?v=BKncta9RQdY

COMPLETE CODE 😃👇


import streamlit as st

import sqlite3

import pandas as pd


conn = sqlite3.connect('railway_system.db')

current_page = 'Login or Sign Up'

c = conn.cursor()



def create_DB_if_Not_available():


    # Create a table to store user information

    c.execute('''CREATE TABLE IF NOT EXISTS users

                (username TEXT, password TEXT)''')


    # Create a table to store employee information

    c.execute('''CREATE TABLE IF NOT EXISTS employees

                (employee_id TEXT, password TEXT, designation TEXT)''')


    # Create a table to store train details

    c.execute('''CREATE TABLE IF NOT EXISTS trains

                (train_number TEXT, train_name TEXT, departure_date TEXT, starting_destination TEXT, ending_destination TEXT)''')

    # return c

create_DB_if_Not_available()


# import sqlite3

# import streamlit as st



def add_train(train_number, train_name, departure_date, starting_destination, ending_destination):

    c.execute("INSERT INTO trains (train_number, train_name, departure_date, starting_destination, ending_destination) VALUES (?, ?, ?, ?, ?)",

              (train_number, train_name, departure_date, starting_destination, ending_destination))

    conn.commit()

    create_seat_table(train_number)



def delete_train(train_number, departure_date):

    # c.execute(f"DROP TABLE seat_{train_number};")

    train_query = c.execute(

        "SELECT * FROM trains WHERE train_number = ?", (train_number,))

    train_data = train_query.fetchone()

    if train_data:

        c.execute("DELETE FROM trains WHERE train_number = ? AND departure_date=?",

                  (train_number, departure_date))


        conn.commit()

        st.success(f"Train with Train Number {train_number} has been deleted.")

    else:

        st.error(f"No such Train with Number {train_number} is available")



conn = sqlite3.connect('railway_system.db')

c = conn.cursor()



def create_seat_table(train_number):

    c.execute(f'''

        CREATE TABLE IF NOT EXISTS seats_{train_number} (

            seat_number INTEGER PRIMARY KEY,

            seat_type TEXT,

            booked INTEGER,

            passenger_name TEXT,

            passenger_age INTEGER,

            passenger_gender TEXT

        )

    ''')


    for i in range(1, 51):

        val = categorize_seat(i)

        c.execute(f'''INSERT INTO seats_{train_number}(seat_number, seat_type, booked, passenger_name, passenger_age, passenger_gender) VALUES (?,?,?,?,?,?);''', (

            i, val, 0, "", "", ""))

    conn.commit()



def categorize_seat(seat_number):

    if (seat_number % 10) in [0, 4, 5, 9]:

        return "Window"

    elif (seat_number % 10) in [2, 3, 6, 7]:

        return "Aisle"

    else:

        return "Middle"



def allocate_next_available_seat(train_number, seat_type):

    # Find the next available seat

    # for seat_number in range(1, 51):

    seat_query = c.execute(

        f"SELECT seat_number FROM seats_{train_number} WHERE booked=0 and seat_type=? ORDER BY seat_number asc", (seat_type,))

    result = seat_query.fetchall()

    # print(result)

    if result:

        # if result  :

        # seat_number = categorize_seat(result[0])

        return result[0]

    # return None, None



def book_ticket(train_number, passenger_name, passenger_age, passenger_gender, seat_type):

    train_query = c.execute(

        "SELECT * FROM trains WHERE train_number = ?", (train_number,))

    train_data = train_query.fetchone()

    if train_data:

        seat_number = allocate_next_available_seat(train_number, seat_type)

        if seat_number:

            # Update the seat as booked and store passenger details

            c.execute(f"UPDATE seats_{train_number} SET booked=1, seat_type=?, passenger_name=?, passenger_age=?, passenger_gender=? WHERE seat_number=?", (

                seat_type, passenger_name, passenger_age, passenger_gender, seat_number[0]))

            conn.commit()

            st.success(

                f"Successfully booked seat {seat_number[0]} ({seat_type}) for {passenger_name}.")

        else:

            st.error("No available seats for booking in this train.")

    else:

        st.error(f"No such Train with Number {train_number} is available")



def cancel_tickets(train_number, seat_number):

    train_query = c.execute(

        "SELECT * FROM trains WHERE train_number = ?", (train_number,))

    train_data = train_query.fetchone()

    if train_data:

        c.execute(

            f'''UPDATE seats_{train_number} SET booked=0, passenger_name='', passenger_age='', passenger_gender='' WHERE seat_number=?''', (seat_number,))

        conn.commit()

        st.success(

            f"Successfully booked seat {seat_number} from {train_number} .")

    else:

        st.error(f"No such Train with Number {train_number} is available")



def search_train_by_train_number(train_number):

    train_query = c.execute(

        "SELECT * FROM trains WHERE train_number = ?", (train_number,))

    train_data = train_query.fetchone()

    return train_data



def search_trains_by_destinations(starting_destination, ending_destination):

    train_query = c.execute("SELECT * FROM trains WHERE starting_destination = ? AND ending_destination = ?",

                            (starting_destination, ending_destination))

    train_data = train_query.fetchall()

    return train_data



def view_seats(train_number):

    train_query = c.execute(

        "SELECT * FROM trains WHERE train_number = ?", (train_number,))

    train_data = train_query.fetchone()

    if train_data:

        # return train_data

        seat_query = c.execute(

            f'''SELECT 'Number : ' || seat_number, '\n Type : '  || seat_type ,'\n Name : ' ||  passenger_name , '\n Age : ' || passenger_age ,'\n Gender : ' ||  passenger_gender as Details, booked  FROM seats_{train_number} ORDER BY seat_number asc''')

        result = seat_query.fetchall()

        if result:

            st.dataframe(data=result)

    else:

        st.error(f"No such Train with Number {train_number} is available")



def train_functions():

    st.title("Train Administrator")

    functions = st.sidebar.selectbox("Select Train Functions", [

        "Add Train", "View Trains", "Search Train", "Delete Train", "Book Ticket", "Cancel Ticket", "View Seats"])

    if functions == "Add Train":

        st.header("Add New Train")

        with st.form(key='new_train_details'):

            train_number = st.text_input("Train Number")

            train_name = st.text_input("Train Name")

            departure_date = st.date_input("Date of Departure")

            starting_destination = st.text_input("Starting Destination")

            ending_destination = st.text_input("Ending Destination")

            submitted = st.form_submit_button("Add Train")

        if submitted and train_name != "" and train_number != '' and starting_destination != "" and ending_destination != "":

            # Insert the new train into the database and create seat table

            add_train(train_number, train_name, departure_date,

                      starting_destination, ending_destination)

            st.success("Train Added Successfully!")

    elif functions == "View Trains":

        st.title("View All Trains")

        # Query all available trains from the database

        train_query = c.execute("SELECT * FROM trains")

        trains = train_query.fetchall()


        if trains:

            st.header("Available Trains:")

            st.dataframe(data=trains)

        else:

            st.error("No trains available in the database.")

    elif functions == "Search Train":

        st.title("Train Details Search")


        st.write("Search by Train Number:")

        train_number = st.text_input("Enter Train Number:")


        st.write("Search by Starting and Ending Destination:")

        starting_destination = st.text_input("Starting Destination:")

        ending_destination = st.text_input("Ending Destination:")


        if st.button("Search by Train Number"):

            if train_number:

                train_data = search_train_by_train_number(train_number)

                if train_data:

                    st.header("Search Result:")

                    st.table(pd.DataFrame([train_data], columns=[

                        "Train Number", "Train Name", "Departure Date", "Starting Destination", "Ending Destination"]))

                else:

                    st.error(

                        f"No train found with the train number: {train_number}")


        if st.button("Search by Destinations"):

            if starting_destination and ending_destination:

                train_data = search_trains_by_destinations(

                    starting_destination, ending_destination)

                if train_data:

                    st.header("Search Results:")

                    df = pd.DataFrame(train_data, columns=[

                        "Train Number", "Train Name", "Departure Date", "Starting Destination", "Ending Destination"])

                    st.table(df)

                else:

                    st.error(

                        f"No trains found for the given source and destination.")


    elif functions == "Delete Train":

        st.title("Delete Train")

        train_number = st.text_input("Enter Train Number to delete:")

        departure_date = st.date_input("Enter the Train Departure date")

        if st.button("Delete Train"):

            if train_number:

                c.execute(f"DROP TABLE IF EXISTS seats_{train_number}")

                delete_train(train_number, departure_date)

    elif functions == "Book Ticket":

        st.title("Book Train Ticket")

        train_number = st.text_input("Enter Train Number:")

        seat_type = st.selectbox(

            "Seat Type", ["Aisle", "Middle", "Window"], index=0)

        # if seat_type=""

        passenger_name = st.text_input("Passenger Name")

        passenger_age = st.number_input("Passenger Age", min_value=1)

        passenger_gender = st.selectbox(

            "Passenger Gender", ["Male", "Female", "Other"], index=0)


        if st.button("Book Ticket"):

            if train_number and passenger_name and passenger_age and passenger_gender:

                book_ticket(train_number, passenger_name,

                            passenger_age, passenger_gender, seat_type)

    elif functions == "Cancel Ticket":

        st.title("Cancel Ticket")

        train_number = st.text_..............


FULL CODE AND SOURCES 😃👇

https://lkfms.pro/railwaysourcecode


FOLLOW US :


at

No comments:

Post a Comment

Subscribe to: Post Comments (Atom)

SQL INJECTION DETECTION USING RANDOM FOREST CLASSIFIER

  SQL INJECTION DETECTION USING RANDOM FOREST CLASSIFIER