fken0.1.5源代码

fken是一个帮助用户背单词的Linux小工具，使用Python实现。

fken最大的特点是它能让用户一边用Linux终端，一边回顾用户希望记忆的内容，比如单词，名人名言等。

更多关于fken的介绍见 fken0.1.5发布：让Linux终端帮你背单词！

 

fken:

#!/usr/bin/env python #!encoding:utf8 import sys from fken import FkenClient if __name__ == "__main__": client = FkenClient() l = len(sys.argv) try: if l == 1: res = client.do([]) else: res = client.do(sys.argv[1:]) except Exception as e: res = 'fken encountered a error: ' + str(e) finally: print res

 

fkend:

#!/usr/bin/env python #!encoding:utf8 ''' Modified from http://www.jejik.com/articles/2007/02/a_simple_unix_linux_daemon_in_python/ ''' import sys, time from fken import FkenDaemon if __name__ == "__main__": daemon = FkenDaemon('/tmp/fkend.pid') if len(sys.argv) == 2: if 'start' == sys.argv[1]: daemon.start() elif 'stop' == sys.argv[1]: daemon.stop() elif 'restart' == sys.argv[1]: daemon.restart() else: print "Unknown command" sys.exit(2) sys.exit(0) else: print "usage: %s start|stop|restart" % sys.argv[0] sys.exit(2)

 

fken.py：

#!/usr/bin/env python #!encoding:utf8 from daemon import Daemon from socket import * import os, glob import random # Set the socket parameters HOST = "localhost" PORT = 21567 BUF = 1024 ADDR = (HOST,PORT) # Set other constant MAX_NUM_ITEMS = 5 MAX_LEN_RES = 45 FKEN_CONF = "~/.fken.conf" class FkenClient(object): def __init__(self): self._buf = BUF self._addr = ADDR def do(self, argv): # Create socket UDPSock = socket(AF_INET,SOCK_DGRAM) UDPSock.settimeout(0.5) # timeout is 500ms reqData = "" # Format command if len(argv) == 0: cmd = "-g" elif len(argv) == 1 and argv[0] == '--help' : client_help_info = / "/tfken(a.k.a, fuck-english) is a system to help you remember stuffs, like english words, famous saying, poems and so on. fken consists of three parts, fken(client), fkend(server) and fken_prompt_enhance(integrate fken into bash shell)./n/n" + / "/tUsage: fken [OPTIONS]/n" + / "/t/t-g, --get Get a item from fken randomly/n" + / "/t/t-u, --update Update fken according to data files/n" + / "/t/t--get-size Get the total size of data stored in fken/n" + / "/t/t--get-items-num Get the total number of items stored in fken/n/n" + / "/tAuthor: Hongliang TIAN (tatetian@gmail.com)/n" + / "/tDate: Sep 25th, 2009" return client_help_info else: cmd = " ".join(argv) # Send request if(UDPSock.sendto(cmd, self._addr)): try: res, addr = UDPSock.recvfrom(self._buf) except Exception as e: res = str(e) finally: return res else: return 'failed' class FkenServer(object): def __init__(self, dataFileDir): self._addr = ADDR self._buf = BUF self._db = FkenDB(dataFileDir) def run(self): # Create socket and bind to address UDPSock = socket(AF_INET,SOCK_DGRAM) UDPSock.bind(self._addr) # Receive messages while True: cmd,addr = UDPSock.recvfrom(self._buf) res = self.handleCommand(cmd) UDPSock.sendto(res, addr) # Close socket UDPSock.close() def handleCommand(self, cmd): tokens = cmd.split(" /t") l = len(tokens) res = 'unexpected command' if l == 1: if tokens[0] == '--get' or tokens[0] == '-g': res = self.handleCmdGet() elif tokens[0] == '--update' or tokens[0] == '-u': res = self.handleCmdUpdate() elif tokens[0] == '--get-size': res = self.handleCmdGetSize() elif tokens[0] == '--get-items-num': res = self.handleCmdGetItemsNum() return res def handleCmdGet(self): res = self._db.get() return res def handleCmdUpdate(self): res = self._db.update() return "Update succeded" def handleCmdGetSize(self): res = self._db.getSize() return "Currently there're " + str(res/1024) + "kb data in total." def handleCmdGetItemsNum(self): res = self._db.getItemsNum() return "Currently there're " + str(res) + " items in fken storage." class FkenDaemon(Daemon): def __init__(self, pidfile, stdin='/dev/null', stdout='/dev/null', stderr='/dev/null') : Daemon.__init__(self, pidfile, stdin, stdout, stderr) self.loadConf(FKEN_CONF) self._server = FkenServer(self._conf['data_file_dir']) def run(self): self._server.run() def loadConf(self, confFilePath): try: self._conf = {'data_file_dir': None} confFile = open(os.path.expanduser(confFilePath)) for line in confFile.readlines(): # remove the comment part of the line pos = line.find('#') if pos != -1: line = line[0:pos] # parse tokens = line.split('=') if len(tokens) >= 2: # <attr> = <val> key = tokens[0].strip() if self._conf.has_key(key): self._conf[key] = "".join(tokens[1:]).strip() elif len(tokens) == 0: # empty line continue else: # not valid raise Exception("format error") confFile.close() for c in self._conf.keys(): if self._conf[c] == None: raise Exception("attribute "+c+" is missing") except Exception, e: raise Exception("FkenDaemon failed to load conf file: "+str(e)) class FkenDB(object): def __init__(self, dataFileDir): self._dataFileDir = os.path.expanduser(dataFileDir) self.update() def update(self): self.updateFrom(self._dataFileDir) def updateFrom(self, dataFileDir): self._data = [] for path in glob.glob(os.path.join(dataFileDir, "*.dat")): if path == None: continue f = open(path) for l in f.readlines(): l = l.strip() if len(l) != 0: self._data.append(l) f.close() def get(self): l = len(self._data) if l == 0: return "" choice = random.randrange(0, l) res = "" for i in range(MAX_NUM_ITEMS): res += " *" res += self._data[(choice + i) % l] # in case the index overflow if len(res) > MAX_LEN_RES: break return res def getItemsNum(self): return len(self._data) def getSize(self): size = 0 for i in self._data: size += len(i) return size

 

fken_prompt_enhance：

#!/bin/bash export PS1="${debian_chroot:+($debian_chroot)}/n/`if [ /$? = 0 ]; then echo /[/e[32/;1m/]^_^/[/e[0m/]; else echo /[/e[31/;1m/]O_O/[/e[0m/]; fi/` /$(fken) /n/[/e[32;1m/](/[/e[37;1m/]/u@/h/[/e[32;1m/])-(/[/e[37;1m/]/t/[/e[32;1m/])-(/[/e[37;1m/]/w/[/e[32;1m/])/n(/[/e[37;1m/]! /!/[/e[32;1m/])-> /[/e[m/]"

 

daemon.py：

#!/usr/bin/env python #!encoding:utf8 ''' From Python Cookbook http://www.jejik.com/articles/2007/02/a_simple_unix_linux_daemon_in_python/ ''' import sys, os, time, atexit from signal import SIGTERM class Daemon: """ A generic daemon class. Usage: subclass the Daemon class and override the run() method """ def __init__(self, pidfile, stdin='/dev/null', stdout='/dev/null', stderr='/dev/null'): self.stdin = stdin self.stdout = stdout self.stderr = stderr self.pidfile = pidfile def daemonize(self): """ do the UNIX double-fork magic, see Stevens' "Advanced Programming in the UNIX Environment" for details (ISBN 0201563177) http://www.erlenstar.demon.co.uk/unix/faq_2.html#SEC16 """ try: pid = os.fork() if pid > 0: # exit first parent sys.exit(0) except OSError, e: sys.stderr.write("fork #1 failed: %d (%s)/n" % (e.errno, e.strerror)) sys.exit(1) # decouple from parent environment os.chdir("/") os.setsid() os.umask(0) # do second fork try: pid = os.fork() if pid > 0: # exit from second parent sys.exit(0) except OSError, e: sys.stderr.write("fork #2 failed: %d (%s)/n" % (e.errno, e.strerror)) sys.exit(1) # redirect standard file descriptors sys.stdout.flush() sys.stderr.flush() si = file(self.stdin, 'r') so = file(self.stdout, 'a+') se = file(self.stderr, 'a+', 0) os.dup2(si.fileno(), sys.stdin.fileno()) os.dup2(so.fileno(), sys.stdout.fileno()) os.dup2(se.fileno(), sys.stderr.fileno()) # write pidfile atexit.register(self.delpid) pid = str(os.getpid()) file(self.pidfile,'w+').write("%s/n" % pid) def delpid(self): os.remove(self.pidfile) def start(self): """ Start the daemon """ # Check for a pidfile to see if the daemon already runs try: pf = file(self.pidfile,'r') pid = int(pf.read().strip()) pf.close() except IOError: pid = None if pid: message = "pidfile %s already exist. Daemon already running?/n" sys.stderr.write(message % self.pidfile) sys.exit(1) # Start the daemon self.daemonize() self.run() def stop(self): """ Stop the daemon """ # Get the pid from the pidfile try: pf = file(self.pidfile,'r') pid = int(pf.read().strip()) pf.close() except IOError: pid = None if not pid: message = "pidfile %s does not exist. Daemon not running?/n" sys.stderr.write(message % self.pidfile) return # not an error in a restart # Try killing the daemon process try: while 1: os.kill(pid, SIGTERM) time.sleep(0.1) except OSError, err: err = str(err) if err.find("No such process") > 0: if os.path.exists(self.pidfile): os.remove(self.pidfile) else: print str(err) sys.exit(1) def restart(self): """ Restart the daemon """ self.stop() self.start() def run(self): """ You should override this method when you subclass Daemon. It will be called after the process has been daemonized by start() or restart(). """