daviddoji/example-code-2e
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tcp_charfinder.py with async/await

Browse Source
This commit is contained in:
Luciano Ramalho
2015-07-13 23:06:34 -03:00
parent 890495dceb
commit aeee41988e
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
18b-async-await/charfinder/tcp_charfinder.py
View File

@@ -11,12 +11,11 @@ PROMPT = b'?> '
index = UnicodeNameIndex() # <2> index = UnicodeNameIndex() # <2>
@asyncio.coroutine async def handle_queries(reader, writer): # <3>
def handle_queries(reader, writer): # <3>
while True: # <4> while True: # <4>
writer.write(PROMPT) # can't yield from! # <5> writer.write(PROMPT) # can't await! # <5>
yield from writer.drain() # must yield from! # <6> await writer.drain() # must await! # <6>
data = yield from reader.readline() # <7> data = await reader.readline() # <7>
try: try:
query = data.decode().strip() query = data.decode().strip()
except UnicodeDecodeError: # <8> except UnicodeDecodeError: # <8>
@@ -31,7 +30,7 @@ def handle_queries(reader, writer): # <3>
writer.writelines(line.encode() + CRLF for line in lines) # <13> writer.writelines(line.encode() + CRLF for line in lines) # <13>
writer.write(index.status(query, len(lines)).encode() + CRLF) # <14> writer.write(index.status(query, len(lines)).encode() + CRLF) # <14>
yield from writer.drain() # <15> await writer.drain() # <15>
print('Sent {} results'.format(len(lines))) # <16> print('Sent {} results'.format(len(lines))) # <16>
print('Close the client socket') # <17> print('Close the client socket') # <17>