mirror of
https://github.com/emmett-framework/granian.git
synced 2025-08-04 17:08:02 +00:00
c95fd0cba7
* Add lint and format tools and config * Format Python code * Format Rust code * Add lint CI workflow
113 lines
3 KiB
Python
113 lines
3 KiB
Python
import json
|
|
|
|
from granian.rsgi import HTTPProtocol, Scope, WebsocketMessageType, WebsocketProtocol
|
|
|
|
|
|
async def info(scope: Scope, protocol: HTTPProtocol):
|
|
protocol.response_bytes(
|
|
200,
|
|
[('content-type', 'application/json')],
|
|
json.dumps(
|
|
{
|
|
'proto': scope.proto,
|
|
'http_version': scope.http_version,
|
|
'rsgi_version': scope.rsgi_version,
|
|
'scheme': scope.scheme,
|
|
'method': scope.method,
|
|
'path': scope.path,
|
|
'query_string': scope.query_string,
|
|
'headers': dict(scope.headers.items()),
|
|
}
|
|
).encode('utf8'),
|
|
)
|
|
|
|
|
|
async def echo(_, protocol: HTTPProtocol):
|
|
msg = await protocol()
|
|
protocol.response_bytes(200, [('content-type', 'text/plain; charset=utf-8')], msg)
|
|
|
|
|
|
async def echo_stream(_, protocol: HTTPProtocol):
|
|
trx = protocol.response_stream(200, [('content-type', 'text/plain; charset=utf-8')])
|
|
async for msg in protocol:
|
|
await trx.send_bytes(msg)
|
|
|
|
|
|
async def stream(_, protocol: HTTPProtocol):
|
|
trx = protocol.response_stream(200, [('content-type', 'text/plain; charset=utf-8')])
|
|
for _ in range(0, 3):
|
|
await trx.send_bytes(b'test')
|
|
|
|
|
|
async def ws_reject(_, protocol: WebsocketProtocol):
|
|
protocol.close(403)
|
|
|
|
|
|
async def ws_info(scope: Scope, protocol: WebsocketProtocol):
|
|
trx = await protocol.accept()
|
|
|
|
await trx.send_str(
|
|
json.dumps(
|
|
{
|
|
'proto': scope.proto,
|
|
'http_version': scope.http_version,
|
|
'rsgi_version': scope.rsgi_version,
|
|
'scheme': scope.scheme,
|
|
'method': scope.method,
|
|
'path': scope.path,
|
|
'query_string': scope.query_string,
|
|
'headers': dict(scope.headers.items()),
|
|
}
|
|
)
|
|
)
|
|
while True:
|
|
message = await trx.receive()
|
|
if message.kind == WebsocketMessageType.close:
|
|
break
|
|
|
|
protocol.close()
|
|
|
|
|
|
async def ws_echo(_, protocol: WebsocketProtocol):
|
|
trx = await protocol.accept()
|
|
|
|
while True:
|
|
message = await trx.receive()
|
|
if message.kind == WebsocketMessageType.close:
|
|
break
|
|
elif message.kind == WebsocketMessageType.bytes:
|
|
await trx.send_bytes(message.data)
|
|
else:
|
|
await trx.send_str(message.data)
|
|
|
|
protocol.close()
|
|
|
|
|
|
async def ws_push(_, protocol: WebsocketProtocol):
|
|
trx = await protocol.accept()
|
|
|
|
try:
|
|
while True:
|
|
await trx.send_str('ping')
|
|
except Exception:
|
|
pass
|
|
|
|
protocol.close()
|
|
|
|
|
|
async def err_app(scope: Scope, protocol: HTTPProtocol):
|
|
1 / 0
|
|
|
|
|
|
def app(scope, protocol):
|
|
return {
|
|
'/info': info,
|
|
'/echo': echo,
|
|
'/echos': echo_stream,
|
|
'/stream': stream,
|
|
'/ws_reject': ws_reject,
|
|
'/ws_info': ws_info,
|
|
'/ws_echo': ws_echo,
|
|
'/ws_push': ws_push,
|
|
'/err_app': err_app,
|
|
}[scope.path](scope, protocol)
|