mirror of https://github.com/craigerl/aprsd.git
85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
import argparse
|
|
import logging
|
|
from logging.handlers import RotatingFileHandler
|
|
import socketserver
|
|
import sys
|
|
import time
|
|
|
|
from aprsd import utils
|
|
|
|
|
|
# command line args
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--loglevel",
|
|
default="DEBUG",
|
|
choices=["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG"],
|
|
help="The log level to use for aprsd.log",
|
|
)
|
|
parser.add_argument("--quiet", action="store_true", help="Don't log to stdout")
|
|
|
|
parser.add_argument("--port", default=9099, type=int, help="The port to listen on .")
|
|
parser.add_argument("--ip", default="127.0.0.1", help="The IP to listen on ")
|
|
|
|
CONFIG = None
|
|
LOG = logging.getLogger("ARPSSERVER")
|
|
|
|
|
|
# Setup the logging faciility
|
|
# to disable logging to stdout, but still log to file
|
|
# use the --quiet option on the cmdln
|
|
def setup_logging(args):
|
|
global LOG
|
|
levels = {
|
|
"CRITICAL": logging.CRITICAL,
|
|
"ERROR": logging.ERROR,
|
|
"WARNING": logging.WARNING,
|
|
"INFO": logging.INFO,
|
|
"DEBUG": logging.DEBUG,
|
|
}
|
|
log_level = levels[args.loglevel]
|
|
|
|
LOG.setLevel(log_level)
|
|
log_format = "%(asctime)s [%(threadName)-12.12s] [%(levelname)-5.5s]" " %(message)s"
|
|
date_format = "%m/%d/%Y %I:%M:%S %p"
|
|
log_formatter = logging.Formatter(fmt=log_format, datefmt=date_format)
|
|
fh = RotatingFileHandler("aprs-server.log", maxBytes=(10248576 * 5), backupCount=4)
|
|
fh.setFormatter(log_formatter)
|
|
LOG.addHandler(fh)
|
|
|
|
if not args.quiet:
|
|
sh = logging.StreamHandler(sys.stdout)
|
|
sh.setFormatter(log_formatter)
|
|
LOG.addHandler(sh)
|
|
|
|
|
|
class MyAPRSTCPHandler(socketserver.BaseRequestHandler):
|
|
def handle(self):
|
|
# self.request is the TCP socket connected to the client
|
|
self.data = self.request.recv(1024).strip()
|
|
LOG.debug(f"{self.client_address[0]} wrote:")
|
|
LOG.debug(self.data)
|
|
# just send back the same data, but upper-cased
|
|
self.request.sendall(self.data.upper())
|
|
|
|
|
|
def main():
|
|
global CONFIG
|
|
args = parser.parse_args()
|
|
setup_logging(args)
|
|
LOG.info("Test APRS server starting.")
|
|
time.sleep(1)
|
|
|
|
CONFIG = utils.parse_config(args)
|
|
|
|
ip = CONFIG["aprs"]["host"]
|
|
port = CONFIG["aprs"]["port"]
|
|
LOG.info(f"Start server listening on {args.ip}:{args.port}")
|
|
|
|
with socketserver.TCPServer((ip, port), MyAPRSTCPHandler) as server:
|
|
server.serve_forever()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|