fken0.1.5源代码

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

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

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

 

fken:

#!/usr/bin/env python#!encoding:utf8import sysfrom fken import FkenClientif __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, timefrom fken import FkenDaemonif __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:utf8from daemon import Daemonfrom socket import *import os, globimport random# Set the socket parametersHOST = "localhost"PORT = 21567BUF = 1024ADDR = (HOST,PORT)# Set other constantMAX_NUM_ITEMS = 5MAX_LEN_RES = 45FKEN_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 ([e-mail])/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/bashexport 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 Cookbookhttp://www.jejik.com/articles/2007/02/a_simple_unix_linux_daemon_in_python/'''import sys, os, time, atexitfrom 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 = stdinself.stdout = stdoutself.stderr = stderrself.pidfile = pidfiledef 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 parentsys.exit(0) except OSError, e: sys.stderr.write("fork #1 failed: %d (%s)/n" % (e.errno, e.strerror))sys.exit(1)# decouple from parent environmentos.chdir("/") os.setsid() os.umask(0) # do second forktry: pid = os.fork() if pid > 0:# exit from second parentsys.exit(0) except OSError, e: sys.stderr.write("fork #2 failed: %d (%s)/n" % (e.errno, e.strerror))sys.exit(1) # redirect standard file descriptorssys.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 pidfileatexit.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 runstry:pf = file(self.pidfile,'r')pid = int(pf.read().strip())pf.close()except IOError:pid = Noneif pid:message = "pidfile %s already exist. Daemon already running?/n"sys.stderr.write(message % self.pidfile)sys.exit(1)# Start the daemonself.daemonize()self.run()def stop(self):"""Stop the daemon"""# Get the pid from the pidfiletry:pf = file(self.pidfile,'r')pid = int(pf.read().strip())pf.close()except IOError:pid = Noneif 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 processtry: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 beendaemonized by start() or restart()."""