socket__tcp__examples/http_server.py - master - ilya.petrash/SimplePyScripts

socket__tcp__examples/http_server.py
83 строки2 KB
gil9red
Refactoring. Using black code style
26 июн 2023, 13:24
26 июн 2023, 13:243790fd6
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = "ipetrash"


# SOURCE: http://andreymal.org/socket3/

import time
import socket


def send_answer(conn, status="200 OK", typ="text/plain; charset=utf-8", data=""):
    data = data.encode("utf-8")

    conn.send(b"GET HTTP/1.1 " + status.encode("utf-8") + b"\r\n")
    conn.send(b"Server: simplehttp\r\n")
    conn.send(b"Connection: close\r\n")
    conn.send(b"Content-Type: " + typ.encode("utf-8") + b"\r\n")
    conn.send(b"Content-Length: " + str(len(data)).encode() + b"\r\n")
    conn.send(b"\r\n")  # После пустой строки в HTTP начинаются данные
    conn.send(data)


def parse(conn):  # Обработка соединения в отдельной функции
    data = b""

    while b"\r\n" not in data:  # Ждём первую строку
        tmp = conn.recv(1024)

        # Сокет закрыли, пустой объект
        if not tmp:
            break
        else:
            data += tmp

    # Данные не пришли
    if not data:
        return

    udata = data.decode("utf-8")

    # берём только первую строку
    udata = udata.split("\r\n", 1)[0]
    # разбиваем по пробелам нашу строку
    method, address, protocol = udata.split(" ", 2)

    if method != "GET" or address != "/time.html":
        send_answer(conn, "404 Not Found", data="Не найдено")
        return

    answer = """<!DOCTYPE html>"""
    answer += """<html><head><title>Время</title></head><body><h1>"""
    answer += time.strftime("%H:%M:%S %d.%m.%Y")
    answer += """</h1></body></html>"""

    send_answer(conn, typ="text/html; charset=utf-8", data=answer)


PORT = 8080

sock = socket.socket()
sock.bind(("", PORT))
sock.listen()

print(f"HTTP server running on http://127.0.0.1:{PORT}/time.html")

try:
    # Работаем постоянно
    while True:
        conn, addr = sock.accept()
        print(f"New connection from {addr[0]}:{addr[1]}")

        try:
            parse(conn)
        except:
            send_answer(conn, "500 Internal Server Error", data="Ошибка")

        finally:
            conn.close()

finally:
    sock.close()