minimal visual improvements

vunerability added
table creation added
2025-12-15 23:07:46 +10:00 · 2025-12-15 23:05:11 +10:00 · 2025-12-15 22:44:12 +10:00
3 changed files with 140 additions and 4 deletions
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 build
 .vscode
 **__pycache__
 data
--- a/readme.md
+++ b/readme.md
@@ -1 +1,10 @@
 К сожалениюя я не любитель PHP, так что всё будет на питоне (Fastapi|Uvicorn)
 **На сервере Web написать уязвимое веб-приложение Весь проект писать на рабочем столе в папке Фамилия_Web_app, где “Фамилия” - это ваша настоящая фамилия**
 1. Реализовать страницу регистрации пользователей **(1.5 балла)**
 2. Реализовать страницу авторизации пользователей **(1.5 балла)**
 3. Реализовать **Cookie (2 балла)**
 4. Реализовать страницу **“приветствия пользователя”** (она должна выводить текст **“Привет, имя_пользователя”**)  (Имя_пользователя, это переменная и должна равняться авторизованному пользователю), доступ на эту страницу только после авторизации **(2 балла)**
 5. Обернуть готовое приложение в **Docker (3 балла)**
--- a/src/app.py
+++ b/src/app.py
@@ -1,8 +1,13 @@
 import os 
 import sqlite3
 from fastapi import FastAPI
 from fastapi import Request, Form
 from fastapi.responses import HTMLResponse, RedirectResponse
 DB_PATH = "./data/database.db"
 conn = sqlite3.connect(DB_PATH)
 cursor = conn.cursor()
 cursor.execute("CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY AUTOINCREMENT, login TEXT NOT NULL UNIQUE, password TEXT NOT NULL)")
 app = FastAPI(
    docs_url=None,      # Disable Swagger UI
@@ -10,7 +15,128 @@ app = FastAPI(
    openapi_url=None    # Disable OpenAPI JSON schema
 )
 STYLES = """
 <style>
 body {
    display: flex;
    flex-direction: column;
    align-items: center;
    justify-content: center;
    min-height: 100vh;
 }
 </style>
 """
-@app.get("/")
+@app.get("/register", response_class=HTMLResponse)
 async def register_form():
    return f"""
    <html>
    <head><title>Регистрация</title>{STYLES}</head>
    <body>
    <h2>Регистрация</h2>
    <form action="/register" method="post">
        <input name="login" placeholder="Login" required><br>
        <input name="password" type="password" placeholder="Password" required><br>
        <button type="submit">Register</button>
    </form>
    <p>Уже есть аккаунт? <a href="/login">Войти</a></p>
    </body>
    </html>
    """
@app.post("/register")
 async def register(login: str = Form(...), password: str = Form(...)):
    try:
        cursor.execute("INSERT INTO users (login, password) VALUES (?, ?)", (login, password))
        conn.commit()
        response = RedirectResponse(url="/welcome", status_code=302)
        response.set_cookie("login", login)
        response.set_cookie("password", password)        
        return response
    except sqlite3.IntegrityError:
        return HTMLResponse(f"""
        <html>
        <head><title>Ошибка регистрации</title>{STYLES}</head>
        <body>
        <h3>Login уже существует</h3>
        <a href='/register'>Попробовать снова</a>
        </body>
        </html>
        """, status_code=400)
@app.get("/login", response_class=HTMLResponse)
 async def login_form():
    return f"""
    <html>
    <head><title>Вход</title>{STYLES}</head>
    <body>
    <h2>Вход</h2>
    <form action="/login" method="post">
        <input name="login" placeholder="Login" required><br>
        <input name="password" type="password" placeholder="Password" required><br>
        <button type="submit">Login</button>
    </form>
    <p>Нет аккаунта? <a href="/register">Зарегистрироваться</a></p>
    </body>
    </html>
    """
@app.post("/login")
 async def login(login: str = Form(...), password: str = Form(...)):
    cursor.execute("SELECT * FROM users WHERE login=? AND password=?", (login, password))
    user = cursor.fetchone()
    if user:
        response = RedirectResponse(url="/welcome", status_code=302)
        response.set_cookie("login", login)
        response.set_cookie("password", password)
        return response
    else:
        return HTMLResponse(f"""
        <html>
        <head><title>Ошибка входа</title>{STYLES}</head>
        <body>
        <h3>Неверные учетные данные</h3>
        <a href='/login'>Попробовать снова</a>
        </body>
        </html>
        """, status_code=401)
@app.get("/welcome", response_class=HTMLResponse)
 async def welcome(request: Request):
    login = request.cookies.get("login")
    password = request.cookies.get("password")
    if not login or not password:
        return RedirectResponse(url="/login")
    cursor.execute("SELECT * FROM users WHERE login=? AND password=?", (login, password))
    user = cursor.fetchone()
    if user:
        return f"""
        <html>
        <head><title>Добро пожаловать</title>{STYLES}</head>
        <body>
        <h1>Привет, {login}</h1>
        <form action="/logout" method="post" onsubmit="return logoutAlert();">
            <button type="submit">Выйти</button>
        </form>
        <script>
        function logoutAlert() {{
            alert('Вы вышли из аккаунта');
            return true;
        }}
        </script>
        </body>
        </html>
        """
    else:
        return RedirectResponse(url="/login")
@app.post("/logout")
 async def logout():
    response = RedirectResponse(url="/login", status_code=302)
    response.delete_cookie("login")
    response.delete_cookie("password")
    return response
@app.get("/", include_in_schema=False)
 async def root():
-    return {"message": "Hello World"}
+    return RedirectResponse(url="/login")
Author	SHA1	Message	Date
OkunElya	3086d1801d	minimal visual improvements	2025-12-15 23:07:46 +10:00
OkunElya	5c4590ff16	vunerability added	2025-12-15 23:05:11 +10:00
OkunElya	c7d8803795	table creation added	2025-12-15 22:44:12 +10:00