import mysql.connector
import pandas as pd
import os
from random import randint
 
class TrainTicketSystem:
    def __init__(self):
        self.db = mysql.connector.connect(
            host="localhost",
            user="root",
            password="1234"
        )
        self.cursor = self.db.cursor()
        self.initialize_database()
 
    def initialize_database(self):
        try:
            # Create the database if it doesn't exist
            self.cursor.execute("CREATE DATABASE IF NOT EXISTS train_ticket_system")
            self.cursor.execute("USE train_ticket_system")
 
            # Create tables
            queries = [
                "CREATE TABLE IF NOT EXISTS admins (name VARCHAR(100), mo_no VARCHAR(15), address VARCHAR(255), email VARCHAR(100), username VARCHAR(50) PRIMARY KEY, password VARCHAR(50))",
                "CREATE TABLE IF NOT EXISTS customers (name VARCHAR(100), mo_no VARCHAR(15), address VARCHAR(255), email VARCHAR(100), username VARCHAR(50) PRIMARY KEY, password VARCHAR(50))",
                "CREATE TABLE IF NOT EXISTS tickets (pnr BIGINT PRIMARY KEY, class VARCHAR(50), boarding_point VARCHAR(100), quota VARCHAR(50))",
                "CREATE TABLE IF NOT EXISTS train_schedule (id INT AUTO_INCREMENT PRIMARY KEY, train_name VARCHAR(100), source VARCHAR(100), destination VARCHAR(100), time VARCHAR(50))"
            ]
            for query in queries:
                self.cursor.execute(query)
            self.db.commit()
        except mysql.connector.Error as err:
            print("Error initializing database:", err)
 
    def admin_signup(self, name, mo_no, address, email, username, password):
        try:
            self.cursor.execute(
                "INSERT INTO admins (name, mo_no, address, email, username, password) VALUES (%s, %s, %s, %s, %s, %s)",
                (name, mo_no, address, email, username, password)
            )
            self.db.commit()
            print("Admin account created.")
        except mysql.connector.Error as err:
            print("Error:", err)
 
    def customer_signup(self, name, mo_no, address, email, username, password):
        try:
            self.cursor.execute(
                "INSERT INTO customers (name, mo_no, address, email, username, password) VALUES (%s, %s, %s, %s, %s, %s)",
                (name, mo_no, address, email, username, password)
            )
            self.db.commit()
            print("Customer account created.")
        except mysql.connector.Error as err:
            print("Error:", err)
 
    def book_ticket(self):
        ticket_class = input("Enter ticket class (e.g., Sleeper, AC): ")
        boarding_point = input("Enter boarding point: ")
        quota = input("Enter quota (e.g., General, Tatkal): ")
        try:
            pnr = randint(1000000000, 9999999999)
            self.cursor.execute(
                "INSERT INTO tickets (pnr, class, boarding_point, quota) VALUES (%s, %s, %s, %s)",
                (pnr, ticket_class, boarding_point, quota)
            )
            self.db.commit()
            print(f"Ticket booked successfully with PNR: {pnr}")
        except mysql.connector.Error as err:
            print("Error:", err)
 
    def view_tickets(self):
        self.cursor.execute("SELECT * FROM tickets")
        tickets = self.cursor.fetchall()
        if tickets:
            for ticket in tickets:
                print(f"PNR: {ticket[0]}, Class: {ticket[1]}, Boarding Point: {ticket[2]}, Quota: {ticket[3]}")
        else:
            print("No tickets available.")
 
    def add_train_schedule(self):
        train_name = input("Enter train name: ")
        source = input("Enter source station: ")
        destination = input("Enter destination station: ")
        time = input("Enter departure time (HH:MM): ")
        try:
            self.cursor.execute(
                "INSERT INTO train_schedule (train_name, source, destination, time) VALUES (%s, %s, %s, %s)",
                (train_name, source, destination, time)
            )
            self.db.commit()
            print("Train schedule added successfully.")
        except mysql.connector.Error as err:
            print("Error:", err)
 
    def view_train_schedule(self):
        self.cursor.execute("SELECT * FROM train_schedule")
        schedules = self.cursor.fetchall()
        if schedules:
            for schedule in schedules:
                print(f"Train: {schedule[1]}, Source: {schedule[2]}, Destination: { schedule[3]}, Time: {schedule[4]}")
        else:
            print("No train schedules available.")
 
    def export_to_excel(self, table_name):
        try:
            self.cursor.execute(f"SELECT * FROM {table_name}")
            rows = self.cursor.fetchall()
            columns = [desc[0] for desc in self.cursor.description]
 
            df = pd.DataFrame(rows, columns=columns)
            desktop = os.path.join(os.path.expanduser("~"), "Desktop")
            file_path = os.path.join(desktop, f"{table_name}.xlsx")
 
            df.to_excel(file_path, index=False)
            print(f"Data exported to {file_path}")
        except mysql.connector.Error as err:
            print("Error:", err)
        except Exception as ex:
            print("Unexpected error:", ex)
 
# Main Menu
def main():
    system = TrainTicketSystem()
    while True:
        print('\n╔═[Main Menu]══════════════════╗')
        print(' 1. Admin SignUp')
        print(' 2. Customer SignUp')
        print(' 3. Book Ticket')
        print(' 4. View Tickets')
        print(' 5. Add Train Schedule')
        print(' 6. View Train Schedules')
        print(' 7. Export Data to Excel')
        print(' 8. Exit')
        print('╚══════════════════════════════╝')
 
        try:
            choice = int(input("Enter Your Choice: "))
        except ValueError:
            print("Invalid input. Please enter a number.")
            continue
 
        if choice == 1:
            system.admin_signup("Admin1", "1234567890", "Address1", "admin@example.com", "admin", "password")
        elif choice == 2:
            system.customer_signup("Customer1", "0987654321", "Address2", "cust@example.com", "customer", "password")
        elif choice == 3:
            system.book_ticket()
        elif choice == 4:
            system.view_tickets()
        elif choice == 5:
            system.add_train_schedule()
        elif choice == 6:
            system.view_train_schedule()
        elif choice == 7:
            table_name = input("Enter table name to export (admins/customers/tickets/train_schedule): ")
            system.export_to_excel(table_name)
        elif choice == 8:
            print("Exiting... Goodbye!")
            break
        else:
            print("Invalid choice. Please try again.")
 
main()

aW1wb3J0IG15c3FsLmNvbm5lY3RvcgppbXBvcnQgcGFuZGFzIGFzIHBkCmltcG9ydCBvcwpmcm9tIHJhbmRvbSBpbXBvcnQgcmFuZGludAoKY2xhc3MgVHJhaW5UaWNrZXRTeXN0ZW06CiAgICBkZWYgX19pbml0X18oc2VsZik6CiAgICAgICAgc2VsZi5kYiA9IG15c3FsLmNvbm5lY3Rvci5jb25uZWN0KAogICAgICAgICAgICBob3N0PSJsb2NhbGhvc3QiLAogICAgICAgICAgICB1c2VyPSJyb290IiwKICAgICAgICAgICAgcGFzc3dvcmQ9IjEyMzQiCiAgICAgICAgKQogICAgICAgIHNlbGYuY3Vyc29yID0gc2VsZi5kYi5jdXJzb3IoKQogICAgICAgIHNlbGYuaW5pdGlhbGl6ZV9kYXRhYmFzZSgpCgogICAgZGVmIGluaXRpYWxpemVfZGF0YWJhc2Uoc2VsZik6CiAgICAgICAgdHJ5OgogICAgICAgICAgICAjIENyZWF0ZSB0aGUgZGF0YWJhc2UgaWYgaXQgZG9lc24ndCBleGlzdAogICAgICAgICAgICBzZWxmLmN1cnNvci5leGVjdXRlKCJDUkVBVEUgREFUQUJBU0UgSUYgTk9UIEVYSVNUUyB0cmFpbl90aWNrZXRfc3lzdGVtIikKICAgICAgICAgICAgc2VsZi5jdXJzb3IuZXhlY3V0ZSgiVVNFIHRyYWluX3RpY2tldF9zeXN0ZW0iKQoKICAgICAgICAgICAgIyBDcmVhdGUgdGFibGVzCiAgICAgICAgICAgIHF1ZXJpZXMgPSBbCiAgICAgICAgICAgICAgICAiQ1JFQVRFIFRBQkxFIElGIE5PVCBFWElTVFMgYWRtaW5zIChuYW1lIFZBUkNIQVIoMTAwKSwgbW9fbm8gVkFSQ0hBUigxNSksIGFkZHJlc3MgVkFSQ0hBUigyNTUpLCBlbWFpbCBWQVJDSEFSKDEwMCksIHVzZXJuYW1lIFZBUkNIQVIoNTApIFBSSU1BUlkgS0VZLCBwYXNzd29yZCBWQVJDSEFSKDUwKSkiLAogICAgICAgICAgICAgICAgIkNSRUFURSBUQUJMRSBJRiBOT1QgRVhJU1RTIGN1c3RvbWVycyAobmFtZSBWQVJDSEFSKDEwMCksIG1vX25vIFZBUkNIQVIoMTUpLCBhZGRyZXNzIFZBUkNIQVIoMjU1KSwgZW1haWwgVkFSQ0hBUigxMDApLCB1c2VybmFtZSBWQVJDSEFSKDUwKSBQUklNQVJZIEtFWSwgcGFzc3dvcmQgVkFSQ0hBUig1MCkpIiwKICAgICAgICAgICAgICAgICJDUkVBVEUgVEFCTEUgSUYgTk9UIEVYSVNUUyB0aWNrZXRzIChwbnIgQklHSU5UIFBSSU1BUlkgS0VZLCBjbGFzcyBWQVJDSEFSKDUwKSwgYm9hcmRpbmdfcG9pbnQgVkFSQ0hBUigxMDApLCBxdW90YSBWQVJDSEFSKDUwKSkiLAogICAgICAgICAgICAgICAgIkNSRUFURSBUQUJMRSBJRiBOT1QgRVhJU1RTIHRyYWluX3NjaGVkdWxlIChpZCBJTlQgQVVUT19JTkNSRU1FTlQgUFJJTUFSWSBLRVksIHRyYWluX25hbWUgVkFSQ0hBUigxMDApLCBzb3VyY2UgVkFSQ0hBUigxMDApLCBkZXN0aW5hdGlvbiBWQVJDSEFSKDEwMCksIHRpbWUgVkFSQ0hBUig1MCkpIgogICAgICAgICAgICBdCiAgICAgICAgICAgIGZvciBxdWVyeSBpbiBxdWVyaWVzOgogICAgICAgICAgICAgICAgc2VsZi5jdXJzb3IuZXhlY3V0ZShxdWVyeSkKICAgICAgICAgICAgc2VsZi5kYi5jb21taXQoKQogICAgICAgIGV4Y2VwdCBteXNxbC5jb25uZWN0b3IuRXJyb3IgYXMgZXJyOgogICAgICAgICAgICBwcmludCgiRXJyb3IgaW5pdGlhbGl6aW5nIGRhdGFiYXNlOiIsIGVycikKCiAgICBkZWYgYWRtaW5fc2lnbnVwKHNlbGYsIG5hbWUsIG1vX25vLCBhZGRyZXNzLCBlbWFpbCwgdXNlcm5hbWUsIHBhc3N3b3JkKToKICAgICAgICB0cnk6CiAgICAgICAgICAgIHNlbGYuY3Vyc29yLmV4ZWN1dGUoCiAgICAgICAgICAgICAgICAiSU5TRVJUIElOVE8gYWRtaW5zIChuYW1lLCBtb19ubywgYWRkcmVzcywgZW1haWwsIHVzZXJuYW1lLCBwYXNzd29yZCkgVkFMVUVTICglcywgJXMsICVzLCAlcywgJXMsICVzKSIsCiAgICAgICAgICAgICAgICAobmFtZSwgbW9fbm8sIGFkZHJlc3MsIGVtYWlsLCB1c2VybmFtZSwgcGFzc3dvcmQpCiAgICAgICAgICAgICkKICAgICAgICAgICAgc2VsZi5kYi5jb21taXQoKQogICAgICAgICAgICBwcmludCgiQWRtaW4gYWNjb3VudCBjcmVhdGVkLiIpCiAgICAgICAgZXhjZXB0IG15c3FsLmNvbm5lY3Rvci5FcnJvciBhcyBlcnI6CiAgICAgICAgICAgIHByaW50KCJFcnJvcjoiLCBlcnIpCgogICAgZGVmIGN1c3RvbWVyX3NpZ251cChzZWxmLCBuYW1lLCBtb19ubywgYWRkcmVzcywgZW1haWwsIHVzZXJuYW1lLCBwYXNzd29yZCk6CiAgICAgICAgdHJ5OgogICAgICAgICAgICBzZWxmLmN1cnNvci5leGVjdXRlKAogICAgICAgICAgICAgICAgIklOU0VSVCBJTlRPIGN1c3RvbWVycyAobmFtZSwgbW9fbm8sIGFkZHJlc3MsIGVtYWlsLCB1c2VybmFtZSwgcGFzc3dvcmQpIFZBTFVFUyAoJXMsICVzLCAlcywgJXMsICVzLCAlcykiLAogICAgICAgICAgICAgICAgKG5hbWUsIG1vX25vLCBhZGRyZXNzLCBlbWFpbCwgdXNlcm5hbWUsIHBhc3N3b3JkKQogICAgICAgICAgICApCiAgICAgICAgICAgIHNlbGYuZGIuY29tbWl0KCkKICAgICAgICAgICAgcHJpbnQoIkN1c3RvbWVyIGFjY291bnQgY3JlYXRlZC4iKQogICAgICAgIGV4Y2VwdCBteXNxbC5jb25uZWN0b3IuRXJyb3IgYXMgZXJyOgogICAgICAgICAgICBwcmludCgiRXJyb3I6IiwgZXJyKQoKICAgIGRlZiBib29rX3RpY2tldChzZWxmKToKICAgICAgICB0aWNrZXRfY2xhc3MgPSBpbnB1dCgiRW50ZXIgdGlja2V0IGNsYXNzIChlLmcuLCBTbGVlcGVyLCBBQyk6ICIpCiAgICAgICAgYm9hcmRpbmdfcG9pbnQgPSBpbnB1dCgiRW50ZXIgYm9hcmRpbmcgcG9pbnQ6ICIpCiAgICAgICAgcXVvdGEgPSBpbnB1dCgiRW50ZXIgcXVvdGEgKGUuZy4sIEdlbmVyYWwsIFRhdGthbCk6ICIpCiAgICAgICAgdHJ5OgogICAgICAgICAgICBwbnIgPSByYW5kaW50KDEwMDAwMDAwMDAsIDk5OTk5OTk5OTkpCiAgICAgICAgICAgIHNlbGYuY3Vyc29yLmV4ZWN1dGUoCiAgICAgICAgICAgICAgICAiSU5TRVJUIElOVE8gdGlja2V0cyAocG5yLCBjbGFzcywgYm9hcmRpbmdfcG9pbnQsIHF1b3RhKSBWQUxVRVMgKCVzLCAlcywgJXMsICVzKSIsCiAgICAgICAgICAgICAgICAocG5yLCB0aWNrZXRfY2xhc3MsIGJvYXJkaW5nX3BvaW50LCBxdW90YSkKICAgICAgICAgICAgKQogICAgICAgICAgICBzZWxmLmRiLmNvbW1pdCgpCiAgICAgICAgICAgIHByaW50KGYiVGlja2V0IGJvb2tlZCBzdWNjZXNzZnVsbHkgd2l0aCBQTlI6IHtwbnJ9IikKICAgICAgICBleGNlcHQgbXlzcWwuY29ubmVjdG9yLkVycm9yIGFzIGVycjoKICAgICAgICAgICAgcHJpbnQoIkVycm9yOiIsIGVycikKCiAgICBkZWYgdmlld190aWNrZXRzKHNlbGYpOgogICAgICAgIHNlbGYuY3Vyc29yLmV4ZWN1dGUoIlNFTEVDVCAqIEZST00gdGlja2V0cyIpCiAgICAgICAgdGlja2V0cyA9IHNlbGYuY3Vyc29yLmZldGNoYWxsKCkKICAgICAgICBpZiB0aWNrZXRzOgogICAgICAgICAgICBmb3IgdGlja2V0IGluIHRpY2tldHM6CiAgICAgICAgICAgICAgICBwcmludChmIlBOUjoge3RpY2tldFswXX0sIENsYXNzOiB7dGlja2V0WzFdfSwgQm9hcmRpbmcgUG9pbnQ6IHt0aWNrZXRbMl19LCBRdW90YToge3RpY2tldFszXX0iKQogICAgICAgIGVsc2U6CiAgICAgICAgICAgIHByaW50KCJObyB0aWNrZXRzIGF2YWlsYWJsZS4iKQoKICAgIGRlZiBhZGRfdHJhaW5fc2NoZWR1bGUoc2VsZik6CiAgICAgICAgdHJhaW5fbmFtZSA9IGlucHV0KCJFbnRlciB0cmFpbiBuYW1lOiAiKQogICAgICAgIHNvdXJjZSA9IGlucHV0KCJFbnRlciBzb3VyY2Ugc3RhdGlvbjogIikKICAgICAgICBkZXN0aW5hdGlvbiA9IGlucHV0KCJFbnRlciBkZXN0aW5hdGlvbiBzdGF0aW9uOiAiKQogICAgICAgIHRpbWUgPSBpbnB1dCgiRW50ZXIgZGVwYXJ0dXJlIHRpbWUgKEhIOk1NKTogIikKICAgICAgICB0cnk6CiAgICAgICAgICAgIHNlbGYuY3Vyc29yLmV4ZWN1dGUoCiAgICAgICAgICAgICAgICAiSU5TRVJUIElOVE8gdHJhaW5fc2NoZWR1bGUgKHRyYWluX25hbWUsIHNvdXJjZSwgZGVzdGluYXRpb24sIHRpbWUpIFZBTFVFUyAoJXMsICVzLCAlcywgJXMpIiwKICAgICAgICAgICAgICAgICh0cmFpbl9uYW1lLCBzb3VyY2UsIGRlc3RpbmF0aW9uLCB0aW1lKQogICAgICAgICAgICApCiAgICAgICAgICAgIHNlbGYuZGIuY29tbWl0KCkKICAgICAgICAgICAgcHJpbnQoIlRyYWluIHNjaGVkdWxlIGFkZGVkIHN1Y2Nlc3NmdWxseS4iKQogICAgICAgIGV4Y2VwdCBteXNxbC5jb25uZWN0b3IuRXJyb3IgYXMgZXJyOgogICAgICAgICAgICBwcmludCgiRXJyb3I6IiwgZXJyKQoKICAgIGRlZiB2aWV3X3RyYWluX3NjaGVkdWxlKHNlbGYpOgogICAgICAgIHNlbGYuY3Vyc29yLmV4ZWN1dGUoIlNFTEVDVCAqIEZST00gdHJhaW5fc2NoZWR1bGUiKQogICAgICAgIHNjaGVkdWxlcyA9IHNlbGYuY3Vyc29yLmZldGNoYWxsKCkKICAgICAgICBpZiBzY2hlZHVsZXM6CiAgICAgICAgICAgIGZvciBzY2hlZHVsZSBpbiBzY2hlZHVsZXM6CiAgICAgICAgICAgICAgICBwcmludChmIlRyYWluOiB7c2NoZWR1bGVbMV19LCBTb3VyY2U6IHtzY2hlZHVsZVsyXX0sIERlc3RpbmF0aW9uOiB7IHNjaGVkdWxlWzNdfSwgVGltZToge3NjaGVkdWxlWzRdfSIpCiAgICAgICAgZWxzZToKICAgICAgICAgICAgcHJpbnQoIk5vIHRyYWluIHNjaGVkdWxlcyBhdmFpbGFibGUuIikKCiAgICBkZWYgZXhwb3J0X3RvX2V4Y2VsKHNlbGYsIHRhYmxlX25hbWUpOgogICAgICAgIHRyeToKICAgICAgICAgICAgc2VsZi5jdXJzb3IuZXhlY3V0ZShmIlNFTEVDVCAqIEZST00ge3RhYmxlX25hbWV9IikKICAgICAgICAgICAgcm93cyA9IHNlbGYuY3Vyc29yLmZldGNoYWxsKCkKICAgICAgICAgICAgY29sdW1ucyA9IFtkZXNjWzBdIGZvciBkZXNjIGluIHNlbGYuY3Vyc29yLmRlc2NyaXB0aW9uXQoKICAgICAgICAgICAgZGYgPSBwZC5EYXRhRnJhbWUocm93cywgY29sdW1ucz1jb2x1bW5zKQogICAgICAgICAgICBkZXNrdG9wID0gb3MucGF0aC5qb2luKG9zLnBhdGguZXhwYW5kdXNlcigifiIpLCAiRGVza3RvcCIpCiAgICAgICAgICAgIGZpbGVfcGF0aCA9IG9zLnBhdGguam9pbihkZXNrdG9wLCBmInt0YWJsZV9uYW1lfS54bHN4IikKCiAgICAgICAgICAgIGRmLnRvX2V4Y2VsKGZpbGVfcGF0aCwgaW5kZXg9RmFsc2UpCiAgICAgICAgICAgIHByaW50KGYiRGF0YSBleHBvcnRlZCB0byB7ZmlsZV9wYXRofSIpCiAgICAgICAgZXhjZXB0IG15c3FsLmNvbm5lY3Rvci5FcnJvciBhcyBlcnI6CiAgICAgICAgICAgIHByaW50KCJFcnJvcjoiLCBlcnIpCiAgICAgICAgZXhjZXB0IEV4Y2VwdGlvbiBhcyBleDoKICAgICAgICAgICAgcHJpbnQoIlVuZXhwZWN0ZWQgZXJyb3I6IiwgZXgpCgojIE1haW4gTWVudQpkZWYgbWFpbigpOgogICAgc3lzdGVtID0gVHJhaW5UaWNrZXRTeXN0ZW0oKQogICAgd2hpbGUgVHJ1ZToKICAgICAgICBwcmludCgnXG7ilZTilZBbTWFpbiBNZW51XeKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVlycpCiAgICAgICAgcHJpbnQoJyAxLiBBZG1pbiBTaWduVXAnKQogICAgICAgIHByaW50KCcgMi4gQ3VzdG9tZXIgU2lnblVwJykKICAgICAgICBwcmludCgnIDMuIEJvb2sgVGlja2V0JykKICAgICAgICBwcmludCgnIDQuIFZpZXcgVGlja2V0cycpCiAgICAgICAgcHJpbnQoJyA1LiBBZGQgVHJhaW4gU2NoZWR1bGUnKQogICAgICAgIHByaW50KCcgNi4gVmlldyBUcmFpbiBTY2hlZHVsZXMnKQogICAgICAgIHByaW50KCcgNy4gRXhwb3J0IERhdGEgdG8gRXhjZWwnKQogICAgICAgIHByaW50KCcgOC4gRXhpdCcpCiAgICAgICAgcHJpbnQoJ+KVmuKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVkOKVnScpCgogICAgICAgIHRyeToKICAgICAgICAgICAgY2hvaWNlID0gaW50KGlucHV0KCJFbnRlciBZb3VyIENob2ljZTogIikpCiAgICAgICAgZXhjZXB0IFZhbHVlRXJyb3I6CiAgICAgICAgICAgIHByaW50KCJJbnZhbGlkIGlucHV0LiBQbGVhc2UgZW50ZXIgYSBudW1iZXIuIikKICAgICAgICAgICAgY29udGludWUKCiAgICAgICAgaWYgY2hvaWNlID09IDE6CiAgICAgICAgICAgIHN5c3RlbS5hZG1pbl9zaWdudXAoIkFkbWluMSIsICIxMjM0NTY3ODkwIiwgIkFkZHJlc3MxIiwgImFkbWluQGV4YW1wbGUuY29tIiwgImFkbWluIiwgInBhc3N3b3JkIikKICAgICAgICBlbGlmIGNob2ljZSA9PSAyOgogICAgICAgICAgICBzeXN0ZW0uY3VzdG9tZXJfc2lnbnVwKCJDdXN0b21lcjEiLCAiMDk4NzY1NDMyMSIsICJBZGRyZXNzMiIsICJjdXN0QGV4YW1wbGUuY29tIiwgImN1c3RvbWVyIiwgInBhc3N3b3JkIikKICAgICAgICBlbGlmIGNob2ljZSA9PSAzOgogICAgICAgICAgICBzeXN0ZW0uYm9va190aWNrZXQoKQogICAgICAgIGVsaWYgY2hvaWNlID09IDQ6CiAgICAgICAgICAgIHN5c3RlbS52aWV3X3RpY2tldHMoKQogICAgICAgIGVsaWYgY2hvaWNlID09IDU6CiAgICAgICAgICAgIHN5c3RlbS5hZGRfdHJhaW5fc2NoZWR1bGUoKQogICAgICAgIGVsaWYgY2hvaWNlID09IDY6CiAgICAgICAgICAgIHN5c3RlbS52aWV3X3RyYWluX3NjaGVkdWxlKCkKICAgICAgICBlbGlmIGNob2ljZSA9PSA3OgogICAgICAgICAgICB0YWJsZV9uYW1lID0gaW5wdXQoIkVudGVyIHRhYmxlIG5hbWUgdG8gZXhwb3J0IChhZG1pbnMvY3VzdG9tZXJzL3RpY2tldHMvdHJhaW5fc2NoZWR1bGUpOiAiKQogICAgICAgICAgICBzeXN0ZW0uZXhwb3J0X3RvX2V4Y2VsKHRhYmxlX25hbWUpCiAgICAgICAgZWxpZiBjaG9pY2UgPT0gODoKICAgICAgICAgICAgcHJpbnQoIkV4aXRpbmcuLi4gR29vZGJ5ZSEiKQogICAgICAgICAgICBicmVhawogICAgICAgIGVsc2U6CiAgICAgICAgICAgIHByaW50KCJJbnZhbGlkIGNob2ljZS4gUGxlYXNlIHRyeSBhZ2Fpbi4iKQoKbWFpbigp