DB for MSHP LETO CTF READY

db.py

import sqlite3

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

db = conn.cursor()

# query = "CREATE TABLE IF NOT EXISTS users (id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, login TEXT,password TEXT)"

# db.execute(query)

# fake_users = ['krosh','ejik','sovunya','kopatych','barash','pin','losyash','karkarych','nyusha']

# for user in fake_users:
#     query = "INSERT INTO users (login,password) VALUES ('{}','{}')".format(user,user+'password')
#     db.execute(query)



# query = "INSERT INTO users (login,password) VALUES (\"durov\",'7777777')"

# query = "SELECT * FROM users WHERE login = 'durov'"
# 'SELECT id  FROM <table> //УСЛОВИЕ!!!!!'

# db.execute(query)

# results = db.fetchall()

# for x in results:
#     print(x)


login = input('Login: ')
password = input('Password: ')

query = "SELECT * FROM users WHERE login='{}' AND password='{}'".format(login,password)
db.execute(query)

results = db.fetchone()
if results:
    print("Hello,",results[1])
else:
    print("Sorry. Bad password")
conn.commit()