gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-09-06 01:41:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update server.py

Browse Source
This commit is contained in:
Namhyeon Go 2024-03-05 11:32:27 +09:00 committed by GitHub
parent 32c192d524
commit 1df5a10cd6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server.py
View File

@ -468,7 +468,7 @@ def start(): #Main Program
class Extension(): class Extension():
extensions = [] extensions = []
@classmethod @classmethod
def register(cls, f): def register(cls, f):
cls.extensions.append(f) cls.extensions.append(f)
@ -483,14 +483,14 @@ class Extension():
@classmethod @classmethod
def get_rpcmethod(cls, method): def get_rpcmethod(cls, method):
for extension in cls.extension: for extension in cls.extensions:
if extension.type == "rpcmethod" and extension.method == method: if extension.type == "rpcmethod" and extension.method == method:
return extension return extension
return None return None
@classmethod @classmethod
def send_accept(cls, conn, method, success = True): def send_accept(cls, conn, method, success = True):
message = jsonrpc2_encode(f"{method}_accept", { _, message = jsonrpc2_encode(f"{method}_accept", {
"success": success "success": success
}) })
conn.send(message) conn.send(message)
@ -498,8 +498,6 @@ class Extension():
@classmethod @classmethod
def readall(cls, conn): def readall(cls, conn):
data = b'' data = b''
conn.settimeout(1)
while True: while True:
try: try:
chunk = conn.recv(buffer_size) chunk = conn.recv(buffer_size)
@ -508,7 +506,6 @@ class Extension():
data += chunk data += chunk
except: except:
pass pass
return data return data
def __init__(self): def __init__(self):