neo1973-germany/PyTracker/trunk/PyTrackerServer.py

91 lines
2.8 KiB
Python
Raw Normal View History

from __future__ import with_statement
import time
from socket import *
from WriteGPX import *
class TrackServer:
def __init__(self, host, port, hashfile, datadir = '.'):
self.InitSocker(host, port)
self.InitHashdb(hashfile)
self.InitTrackDict()
self.datadir = datadir
def InitHashdb(self, hashfile):
self.hashdb=[]
with open(hashfile, "r") as file:
for line in file:
if line:
self.hashdb.append((line.split()[0], \
line.split()[1]))
def InitTrackDict(self):
self.TrackDict={}
for data in self.hashdb:
self.TrackDict[data[0]] = ""
def InitSocket(self, host, port):
# Set the socket parameters
# e.g. host = "localhost"
# e.g. port = 49152
self.__addr = (str(host),int(port))
# Create socket and bind it to the address
self.__UDPSock = socket(AF_INET,SOCK_DGRAM)
self.__UDPSock.bind(sel.__addr)
# Debug message:
print "UDP Socket for %s at port %s created" % (host, port)
def VerifyUser(self, username, password_hash):
for data in self.hashdb:
if data[0] == username and data[1] == password_hash:
return 1
return 0
def Parser(self, stuff):
# Parses the complete data sent to UDP port
try:
username, password_hash, action, data = stuff.split(';')
# Verifies the user and password
if self.VerifyUser(username, password_hash):
if action == "START":
self.NewTrack(username)
if action == "STOP":
self.CloseTrack(username)
if action == "TRANSMIT":
self.AddToTrack(username, data)
print "Action", action,"received"
except:
print "Something went wrong.."
def NewTrack(self, username):
if not self.TrackDict[username]:
self.TrackDict[username] = WriteGPX("%s%s%s" % (self.datadir, username, time.strftime("%Y%m%d%H%M%S"))
print "Created track", self.TrackDict[username]
def CloseTrack(self, username):
if self.TrackDict[username]:
self.TrackDict[username].close()
print "Closed track", self.TrackDict[username]
del self.TrackDict[username]
def AddToTrack(self, username, data):
lat, lon, ele, time = data.split(',')
self.TrackDict[username].write(lat, lon, ele, time)
buf = 1024
# Create instance of TrackServer
instance = TrackServer()
#Receive messages
while 1:
data,addr = UDPSock.recvfrom(1024)
print "Following data received:", data
instance.Parser(data)