from socketserver import BaseRequestHandler, UDPServer
|
import time
|
import os, sys
|
|
|
cudir,_ = os.path.split(__file__)
|
file_folder = os.path.join(cudir,"saveUDPData")
|
|
if not os.path.exists(file_folder):
|
os.makedirs(file_folder)
|
print(f"make a new folder: {file_folder}")
|
|
packed_count = 0
|
|
class PointCloudUDPRequestHandler(BaseRequestHandler):
|
|
def handle(self):
|
|
global packed_count
|
|
print(time.asctime(),' Got connection from {}, {}'.format(self.client_address, packed_count))
|
# Get message and client socket
|
msg, sock = self.request
|
f = open('{}/packet{}.udp'.format(file_folder,packed_count), 'wb')
|
f.write(msg)
|
f.close()
|
packed_count += 1
|
if packed_count >= 1000000:
|
UDPServer.server_close(server)
|
|
|
if __name__ == '__main__':
|
|
def help():
|
print("Usage: udp_test_get.py [host]. host can be 'test' or 'microwave' ")
|
exit(1)
|
|
if len(sys.argv) == 2:
|
if sys.argv[1] == 'test':
|
host = ''
|
elif sys.argv[1] == 'microwave':
|
host = '192.168.2.65'
|
else:
|
help()
|
else:
|
help()
|
|
global server
|
# host = ''
|
# host = '192.168.2.65' # use this host to get the udp data from microwave device
|
port = 9911
|
server =UDPServer((host, port), PointCloudUDPRequestHandler)
|
print(time.asctime(),'Server started on port', port)
|
print('....')
|
print('ctrl-c to quit server.')
|
try:
|
server.serve_forever()
|
except KeyboardInterrupt:
|
server.server_close()
|
print(time.asctime(),"Server Stopped")
|
except:
|
server.server_close()
|
print(time.asctime(),"Server Stopped")
|
