-
Notifications
You must be signed in to change notification settings - Fork 2
/
Drone-client.py
75 lines (61 loc) · 1.9 KB
/
Drone-client.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
import os
import socket
import subprocess
import time,csv
class Client():
def __int__(self):
self.host='192.168.0.14'
self.port=9999
self.s = None
def socket_create(self):
try:
self.s = socket.socket()
print('Socket Created')
except socket.error as e:
print('Socket creation error: '+ str(e))
def socket_connect(self):
try:
self.s.connect(('192.168.0.14',9999))
print('Socket connected')
except socket.error as e:
print('Socket connection error: '+ str(e))
return
def create_file(self):
filename = str(self.s.recv(1024), 'utf-8')
print('File created:{}'.format(filename))
header = ['Date','Time', 'PM1', 'PM2_5', 'PM10', 'N02', 'CO2', 'CO', 'Humidity', 'Temperature']
with open(filename, 'w+') as file:
writer = csv.writer(file)
writer.writerow(header)
return filename
def recv_data(self, filename):
file = open(filename , 'a', newline='')
print('file opened')
writer = csv.writer(file)
count=0
while True:
print('receiving data...')
data = str(self.s.recv(1024), 'utf-8')
print(data)
if not data:
break
# write data to a file
data = data.split(',')
print(data)
writer.writerow(data)
#file.close()
def main():
client=Client()
client.socket_create()
while True:
try:
client.socket_connect()
except Exception as e:
print('Error on socket connections: '+str(e))
else:
break
filename = client.create_file()
client.recv_data(filename)
client.s.close()
if __name__=='__main__':
main()