kopia lustrzana https://github.com/glidernet/ogn-python
Move ogn client (gateway) to a separate file.
The documentation[1] says: > In the simplest case, __init__.py can just be an empty file, > but it can also execute initialization code for the package or > set the __all__ variable, described later. A class like 'ognGateway' can't be considered 'initialization code'. [1]: https://docs.python.org/3/tutorial/modules.html#packagespull/13/head
rodzic
57f51ec111
commit
6af31fe7a7
|
@ -1,69 +0,0 @@
|
|||
import socket
|
||||
from time import time
|
||||
|
||||
from ogn.gateway import settings
|
||||
from ogn.commands.dbutils import session
|
||||
from ogn.aprs_parser import parse_aprs
|
||||
from ogn.aprs_utils import create_aprs_login
|
||||
from ogn.exceptions import AprsParseError, OgnParseError
|
||||
from ogn.logger import logger
|
||||
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
from ogn.model import Base
|
||||
|
||||
|
||||
class ognGateway:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def connect_db(self):
|
||||
self.session = session
|
||||
|
||||
def connect(self, aprs_user):
|
||||
# create socket, connect to server, login and make a file object associated with the socket
|
||||
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
|
||||
self.sock.connect((settings.APRS_SERVER_HOST, settings.APRS_SERVER_PORT))
|
||||
|
||||
login = create_aprs_login(aprs_user, -1, settings.APRS_APP_NAME, settings.APRS_APP_VER, settings.APRS_FILTER)
|
||||
self.sock.send(login.encode())
|
||||
self.sock_file = self.sock.makefile('rw')
|
||||
|
||||
def disconnect(self):
|
||||
# close everything
|
||||
self.sock.shutdown(0)
|
||||
self.sock.close()
|
||||
|
||||
def run(self):
|
||||
keepalive_time = time()
|
||||
while True:
|
||||
if time() - keepalive_time > settings.APRS_KEEPALIVE_TIME:
|
||||
logger.debug('Sending keepalive')
|
||||
self.sock.send("#keepalive".encode())
|
||||
keepalive_time = time()
|
||||
|
||||
# Read packet string from socket
|
||||
packet_str = self.sock_file.readline().strip()
|
||||
|
||||
# A zero length line should not be return if keepalives are being sent
|
||||
# A zero length line will only be returned after ~30m if keepalives are not sent
|
||||
if len(packet_str) == 0:
|
||||
logger.warning('Read returns zero length string. Failure. Orderly closeout')
|
||||
break
|
||||
|
||||
self.proceed_line(packet_str)
|
||||
|
||||
def proceed_line(self, line):
|
||||
try:
|
||||
beacon = parse_aprs(line)
|
||||
except AprsParseError:
|
||||
logger.error('AprsParseError while parsing line: %s' % line, exc_info=True)
|
||||
return
|
||||
except OgnParseError:
|
||||
logger.error('OgnParseError while parsing line: ' % line, exc_info=True)
|
||||
return
|
||||
|
||||
if beacon is not None:
|
||||
self.session.add(beacon)
|
||||
self.session.commit()
|
|
@ -0,0 +1,69 @@
|
|||
import socket
|
||||
from time import time
|
||||
|
||||
from ogn.gateway import settings
|
||||
from ogn.commands.dbutils import session
|
||||
from ogn.aprs_parser import parse_aprs
|
||||
from ogn.aprs_utils import create_aprs_login
|
||||
from ogn.exceptions import AprsParseError, OgnParseError
|
||||
from ogn.logger import logger
|
||||
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
from ogn.model import Base
|
||||
|
||||
|
||||
class ognGateway:
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
def connect_db(self):
|
||||
self.session = session
|
||||
|
||||
def connect(self, aprs_user):
|
||||
# create socket, connect to server, login and make a file object associated with the socket
|
||||
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
|
||||
self.sock.connect((settings.APRS_SERVER_HOST, settings.APRS_SERVER_PORT))
|
||||
|
||||
login = create_aprs_login(aprs_user, -1, settings.APRS_APP_NAME, settings.APRS_APP_VER, settings.APRS_FILTER)
|
||||
self.sock.send(login.encode())
|
||||
self.sock_file = self.sock.makefile('rw')
|
||||
|
||||
def disconnect(self):
|
||||
# close everything
|
||||
self.sock.shutdown(0)
|
||||
self.sock.close()
|
||||
|
||||
def run(self):
|
||||
keepalive_time = time()
|
||||
while True:
|
||||
if time() - keepalive_time > settings.APRS_KEEPALIVE_TIME:
|
||||
logger.debug('Sending keepalive')
|
||||
self.sock.send("#keepalive".encode())
|
||||
keepalive_time = time()
|
||||
|
||||
# Read packet string from socket
|
||||
packet_str = self.sock_file.readline().strip()
|
||||
|
||||
# A zero length line should not be return if keepalives are being sent
|
||||
# A zero length line will only be returned after ~30m if keepalives are not sent
|
||||
if len(packet_str) == 0:
|
||||
logger.warning('Read returns zero length string. Failure. Orderly closeout')
|
||||
break
|
||||
|
||||
self.proceed_line(packet_str)
|
||||
|
||||
def proceed_line(self, line):
|
||||
try:
|
||||
beacon = parse_aprs(line)
|
||||
except AprsParseError:
|
||||
logger.error('AprsParseError while parsing line: %s' % line, exc_info=True)
|
||||
return
|
||||
except OgnParseError:
|
||||
logger.error('OgnParseError while parsing line: ' % line, exc_info=True)
|
||||
return
|
||||
|
||||
if beacon is not None:
|
||||
self.session.add(beacon)
|
||||
self.session.commit()
|
|
@ -1,6 +1,6 @@
|
|||
import socket
|
||||
|
||||
from ogn.gateway import ognGateway
|
||||
from ogn.gateway.client import ognGateway
|
||||
from ogn.logger import logger
|
||||
|
||||
from manager import Manager
|
||||
|
|
Ładowanie…
Reference in New Issue