-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathserver.py
85 lines (73 loc) · 2.52 KB
/
server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
import socket
import json
from views import *
import ytbscrapper.ytbscrapper
STANDART_URLS = {
'/': root,
'/favicon.ico': favicon
}
address = 'localhost'
port = 5000
package_size = 1024
closing_sequence = b'\r\n\r\n'
def generate_headers(method, url):
if method != 'GET':
return '<body><h1>Method not allowed.</h1></body>', 405
if url in STANDART_URLS:
return STANDART_URLS[url](), 200
video_path, info_path = ytbscrapper.ytbscrapper.get_info(url[1:])
with open(info_path, 'r') as file:
info = str(json.load(file))[:100]
return info, 200
def parse_request(request):
parsed = request.split(' ')
method = parsed[0]
url = parsed[1]
return method, url
def generate_response(request):
method, url = parse_request(request)
print(f'request parsed: {method}, {url}')
headers, code = generate_headers(method, url)
print(f'header generated: {headers}, {code}')
return headers.encode('utf-8')
def listen_clients(connection, client_address):
data_lst = []
while True:
#Цикл отвечающий за приём целого сообщения
try:
buff = connection.recv(package_size)
except ConnectionError:
print(f'{client_address} разорвал соединение. Внутренний цикл.')
break
if not buff:
break
data_lst.append(buff.decode('utf-8'))
if not buff.endswith(closing_sequence):
continue
data = ''.join(data_lst)[:-len(closing_sequence)]
print(data)
data_lst.clear()
#Payload
response = generate_response(data)
print(f'response generated: {response}')
connection.sendall(response)
def run():
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
#Need to delete in prodaction?
server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
server_socket.bind((address, port))
server_socket.listen(1)
print('Server Started.')
while True:
#Цикл отвечающий за работу сервера
try:
connection, client_address = server_socket.accept()
except OSError:
print(f'{client_address} разорвал соединение. Внешний цикл.')
break
listen_clients(connection, client_address)
print('cycle is alive')
#response = generate_response(request.decode('utf-8'))
server_socket.close()
if __name__ == '__main__':
run()