pyterm.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

import cmd, serial, sys, threading, readline, time, ConfigParser, logging, os
from terminal_file import TerminalFile

pytermdir     = os.environ['HOME'] + os.path.sep + '.pyterm'

class SerCmd(cmd.Cmd):

    def __init__(self, port=None):
        cmd.Cmd.__init__(self)
        self.port = port
        self.aliases = dict()
        self.load_config()
        try:
            readline.read_history_file()
        except IOError:
            pass

                ### create Logging object
        my_millis = ("%.4f" % time.time())
        date_str = '%s.%s' % (time.strftime('%Y%m%d-%H:%M:%S'), my_millis[-4:])
        # create formatter
        fmt_str = '%(asctime)s - %(levelname)s # %(message)s'
        formatter = logging.Formatter(fmt_str)
        logging.basicConfig(filename=pytermdir + os.path.sep + date_str + '.log', level=logging.DEBUG, format=fmt_str)
        ch = logging.StreamHandler()
        ch.setLevel(logging.DEBUG)

        # create logger
        self.logger = logging.getLogger('')
        self.logger.setLevel(logging.DEBUG)

        # add formatter to ch
        ch.setFormatter(formatter)
        # add ch to logger
        self.logger.addHandler(ch)


    def preloop(self):
        if not self.port:
            sys.stderr.write("No port specified!\n")
            sys.exit(-1)
        self.ser = serial.Serial(port=self.port, baudrate=115200, dsrdtr=0, rtscts=0)
        self.ser.setDTR(0)
        self.ser.setRTS(0)

        # start serial->console thread
        receiver_thread = threading.Thread(target=reader, args=(self.ser,self.logger))
        receiver_thread.setDaemon(1)
        receiver_thread.start()

    def default(self, line):
        for tok in line.split(';'):
            tok = self.get_alias(tok)
            self.ser.write(tok.strip() + "\n")

    def do_help(self, line):
        self.ser.write("help\n")

    def complete_date(self, text, line, begidx, endidm):
        date = time.strftime("%Y-%m-%d %H:%M:%S")
        return ["%s" % date]

    def do_reset(self, line):
        self.ser.setDTR(1)
        self.ser.setDTR(0)

    def do_exit(self, line):
        readline.write_history_file()
        sys.exit(0)

    def do_save(self, line):
        if not self.config.has_section("general"):
            self.config.add_section("general")
        self.config.set("general", "port", self.port)
        if len(self.aliases):
            if not self.config.has_section("aliases"):
                self.config.add_section("aliases")
            for alias in self.aliases:
                self.config.set("aliases", alias, self.aliases[alias])

        with open(path.expanduser('~/.pyterm'), 'wb') as config_fd:
            self.config.write(config_fd)
            print("Config saved")

    def do_show_config(self, line):
        for key in self.__dict__:
            print(str(key) + ": " + str(self.__dict__[key]))

    def do_alias(self, line):
        if line.endswith("list"):
            for alias in self.aliases:
                print("%s = %s" % (alias, self.aliases[alias]))
            return
        if not line.count("="):
            sys.stderr.write("Usage: alias <ALIAS> = <CMD>\n")
            return
        self.aliases[line.split('=')[0].strip()] = line.split('=')[1].strip()

    def do_rmalias(self, line):
        if not self.aliases.pop(line, None):
            sys.stderr.write("Alias not found")

    def get_alias(self, tok):
        for alias in self.aliases:
            if tok.split()[0] == alias:
                return self.aliases[alias] + tok[len(alias):]
        return tok

    def load_config(self):
        self.config = ConfigParser.SafeConfigParser()
        self.config.read([pytermdir + os.path.sep + 'pyterm.conf'])

        for sec in self.config.sections():
            if sec == "aliases":
                for opt in self.config.options(sec):
                    self.aliases[opt] = self.config.get(sec, opt)
            else:
                for opt in self.config.options(sec):
                    if not self.__dict__.has_key(opt):
                        self.__dict__[opt] = self.config.get(sec, opt)


def reader(ser, logger):
    output = ""
    while (1):
        c = ser.read(1)
        if c == '\n' or c == '\r':
            logger.info(output)
            output = ""
        else:
            output += c
        #sys.stdout.write(c)
        #sys.stdout.flush()

if __name__ == "__main__":
    if not os.path.exists(pytermdir):
        os.makedirs(pytermdir)

    if (len(sys.argv) > 1):
        port = sys.argv[1]
    else:
        port = None

    myshell = SerCmd(port)
    myshell.prompt = ''

    myshell.cmdloop("Welcome to pyterm")