jdz/gps01.py

126 lines
3.2 KiB
Python
Raw Normal View History

2024-07-16 03:00:37 +00:00
#!/usr/bin/python
# -*- coding:utf-8 -*-
import RPi.GPIO as GPIO
import serial
import time, os
import csv
from datetime import datetime
2024-07-23 00:25:16 +00:00
2024-07-22 01:55:14 +00:00
# GPIO configuration
GPIO.setmode(GPIO.BCM)
GPIO.setwarnings(False)
2024-07-25 01:39:15 +00:00
ser = serial.Serial('/dev/ttyUSB3',115200)
2024-07-16 03:00:37 +00:00
ser.flushInput()
rec_buff = ''
time_count = 0
2024-07-22 02:49:33 +00:00
myhost = os.uname()[1]
2024-07-23 00:25:16 +00:00
2024-07-16 03:00:37 +00:00
# CSV file settings
csv_filename = f"gps_data_{myhost}.csv"
csv_headers = ["Timestamp", "Latitude", "Longitude"]
def convert_to_decimal(coord_str):
coord = coord_str[:-2]
direction = coord_str[-1]
2024-07-22 02:49:33 +00:00
2024-07-16 03:00:37 +00:00
degrees = float(coord[:2] if direction in ['N', 'S'] else coord[:3])
minutes = float(coord[2:] if direction in ['N', 'S'] else coord[3:])
decimal = degrees + (minutes / 60)
if direction in ['S', 'W']:
decimal = -decimal
return decimal
def parse_gps_data(gps_string):
parts = gps_string.split(',')
if len(parts) != 4:
return "Invalid GPS data"
lat = convert_to_decimal(parts[0] + ',' + parts[1])
lon = convert_to_decimal(parts[2] + ',' + parts[3])
return f"{lat:.6f}", f"{lon:.6f}"
def write_to_csv(lat, lon):
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
with open(csv_filename, mode='a', newline='') as file:
writer = csv.writer(file)
writer.writerow([timestamp, lat, lon])
2024-07-22 01:55:14 +00:00
2024-07-16 03:00:37 +00:00
def send_at(command,back,timeout):
rec_buff = ''
ser.write((command+'\r\n').encode())
time.sleep(timeout)
if ser.inWaiting():
time.sleep(0.01)
rec_buff = ser.read(ser.inWaiting())
if rec_buff != '':
if back not in rec_buff.decode():
print(command + ' ERROR')
print(command + ' back:\t' + rec_buff.decode())
return 0
else:
gps_data = rec_buff.decode()[13:-36]
print("Raw GPS data:", gps_data)
lat, lon = parse_gps_data(gps_data)
converted_gps = f"{lat}, {lon}"
print("Converted GPS data for Google Maps:", converted_gps)
2024-07-22 01:55:14 +00:00
current_time = datetime.datetime.now()
write_to_csv(current_time, lat, lon)
2024-07-16 03:00:37 +00:00
return 1
else:
print('GPS no está listo')
return 0
def get_gps_position():
rec_null = True
answer = 0
print('Iniciando GPS')
rec_buff = ''
send_at('AT+CGPS=1,1','OK',1)
time.sleep(2)
while rec_null:
answer = send_at('AT+CGPSINFO','+CGPSINFO: ',1)
client.publish("iiot/"+ myhost +"/gps", rec_buff, 0)
if 1 == answer:
answer = 0
if ',,,,,,' in rec_buff:
print('GPS no está listo')
rec_null = False
time.sleep(1)
else:
print('error %d'%answer)
rec_buff = ''
send_at('AT+CGPS=0','OK',1)
return False
time.sleep(1.5)
# Create CSV file with headers if it doesn't exist
2024-07-23 00:25:16 +00:00
if not os.path.exists(csv_filename):
with open(csv_filename, mode='w', newline='') as file:
writer = csv.writer(file)
writer.writerow([timestamp, lat, lon])
writer.writerow(csv_headers)
2024-07-16 03:00:37 +00:00
try:
get_gps_position()
2024-07-22 01:55:14 +00:00
2024-07-16 03:00:37 +00:00
except:
if ser != None:
ser.close()
GPIO.cleanup()
2024-07-22 01:55:14 +00:00
2024-07-16 03:00:37 +00:00
if ser != None:
2024-07-23 00:25:16 +00:00
ser.close()
GPIO.cleanup()