让程序只能启动一份

有时写的程序因为资源等等原因，应该只启动一份。利用指定的文件锁，可以实现这样的功能。

import os
import fcntl
import errno

def lock_file(filename):
    fd = os.open(filename, os.O_CREAT | os.O_WRONLY, 0666)
    try:
        fcntl.lockf(fd, fcntl.LOCK_EX | fcntl.LOCK_NB)
        return True
    except IOError, e:
        if e.errno in (errno.EACCES, errno.EAGAIN):
            return False

if not lock_file('/tmp/test.lock'):
  print "another instance is running"lock_file('/tmp/test.lock')

再附上一份perl的代码：

use Fcntl qw(:flock);

  my $lockdir = 'lock';
  if (!-d $lockdir) {
      mkdir $lockdir, 0755;
      my $status=$!;
      die "Failed to create $lockdir: $status\n" if (!-d $lockdir);
  }
  my $lockfile="$lockdir/test.pid";
  if (!open(PID, ">$lockfile")) {
    die "can not open pid file\n";
  }
  unless (flock(PID, LOCK_EX|LOCK_NB)) {
    die "can not lock pid file\n";
  }
  print "locked\n";

顺便再抱怨两句，perl5补丁摞补丁的语法很怪异，异常处理机制竟然要用if...unless，类的写法也搞得跟写dll一样。不得不说，perl的语法离现代语言太远了。一个优美的语言可以提高开发效率，期待一下perl6最终出来的样子。

在python和perl程序中启用logging记录日志

写一些脚本程序的时候，合理的记录日志是必不可少的，尽量不往stdout乱打印信息为好，这时python的logging模块很用用处。不过调试时为了方便，还是希望日志也打印到stdout一份，这样出现什么问题一目了然；否则就只有再开个terminal，用tail -f my.log来检查了。

import logging

def _init_logging(logfile, debug=False):
    if debug:
        level = logging.DEBUG
    else:
        level = logging.INFO
    logging.basicConfig(level=level, format='%(asctime)s %(message)s',
            datefmt='%Y-%m-%d %H:%M:%S', filename=logfile, filemode='w')
    if debug:
        console = logging.StreamHandler()
        console.setLevel(logging.DEBUG)
        formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(message)s')
        console.setFormatter(formatter)
        logging.getLogger('').addHandler(console)

if __name__ == "__main__":
    import os
    debug = os.environ.get('DEBUG') and True or False
    _init_logging('my.log', debug=debug)
    logging.info('start')

    logging.info('end')

这样程序就会知道检查环境变量DEBUG，往合适的地方打印信息了。调试程序时执行方法：

$ DEBUG=1 ./my.py
2007-10-22 17:30:59,917 INFO     start
2007-10-22 17:31:02,027 INFO     end

最近还写了一些perl代码，和上面差不多功能的perl代码也贴出来：

use strict;
use Log::Log4perl qw(:easy);

my $log_file = "/tmp/my.log";

if (exists $ENV{DEBUG} && $ENV{DEBUG}) {
  Log::Log4perl->easy_init(
    {file  => ">> " . log_file, level => $DEBUG},
    {file  => "STDOUT", level => $DEBUG},
  );
} else {
  Log::Log4perl->easy_init(
    {file  => ">> " . $log_file, level => $DEBUG},
  );
}

INFO("start...");

这里用到log4perl模块，需要提前安装：

sudo perl -MCPAN -e'install Log::Log4perl'

虽然适应了一段时间，但perl满眼$@%这些助记符，还是很阻碍阅读和思维连贯，不习惯。