示例#1
0
 def test_partial_file(self):
     tempdir = tempfile.mkdtemp()
     with open(os.path.join(tempdir, 'timelog'), 'w') as f:
         f.write('i 2010/03/16 12:20:19 misc\n')
         f.write('o 2010/03/16 12:22:19 out\n')
     now = datetime.datetime(2010, 03, 17)
     timelog.clock_in(
         reason='test',
         log_path=os.path.join(tempdir, 'timelog'),
         _now=now,
         )
     with open(os.path.join(tempdir, 'timelog'), 'r') as f:
         lines = f.readlines()
         wanted = [
             'i 2010/03/16 12:20:19 misc\n',
             'o 2010/03/16 12:22:19 out\n',
             ' '.join(
                 [
                     'i',
                     now.strftime("%Y/%m/%d"),
                     now.strftime("%H:%M:%S"),
                     'test\n',
                     ]
                 ),
             ]
         assert lines == wanted
示例#2
0
文件: cli.py 项目: thedaniel/dhclock
def clock():
    parser = OptionParser()
    parser.add_option("-l", "--log", dest="log_path", help="location of timelog file")
    parser.add_option("--c", "--config", dest="config_path", help="path to config file")
    opts, args = parser.parse_args()

    config_path = opts.config_path
    if config_path is None:
        config_path = CONFIG_PATH
    if os.path.exists(config_path):
        conf = ConfigParser()
        conf.read([config_path])
        try:
            log_path = conf.get("dhclock", "log_path")
        except ConfigParser.NoOptionError:
            log_path = opts.log_path  # None if unset
    if log_path is None:
        log_path = LOG_PATH
    if len(args) != 2:
        parser.error("use: clock.py in/out reason")
    if args[0] == "in":
        timelog.clock_in(args[1], log_path)
    elif args[0] == "out":
        timelog.clock_out(args[1], log_path)
    elif args[0] == "switch":
        timelog.clock_out("switch", log_path)
        timelog.clock_in(args[1], log_path)
    else:
        parser.error("use: clock.py in/out/switch reason")
示例#3
0
 def test_empty_file(self):
     tempdir = tempfile.mkdtemp()
     f = open(os.path.join(tempdir, 'timelog'), 'w')
     f.close()
     now = datetime.datetime(2010, 05, 01)
     timelog.clock_in(reason='test', log_path=os.path.join(tempdir, 'timelog'), _now=now)
     assert 'timelog' in os.listdir(tempdir)
     with open(os.path.join(tempdir, 'timelog')) as f:
         lines = f.readlines()
         assert len(lines) == 1
         assert lines[0] == ' '.join(
             [
                 'i',
                 now.strftime("%Y/%m/%d"),
                 now.strftime("%H:%M:%S"),
                 'test\n',
                 ]
             )