#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os.path

import tornado.auth
import tornado.escape
import tornado.httpserver
import tornado.ioloop
import tornado.options
import tornado.web
from tornado.options import define, options

import pymongo
import time
import datetime
import xlrd

define("port", default=8007, help="run on the given port", type=int)

class Application(tornado.web.Application):
  def __init__(self):
    handlers = [
      (r"/", MainHandler),
      (r"/read", ReadHandler),
      (r"/day", DayHandler),
    ]
    settings = dict(
      template_path=os.path.join(os.path.dirname(__file__), "templates"),
      static_path=os.path.join(os.path.dirname(__file__), "static"),
      debug=True,
      )
    conn = pymongo.Connection("localhost", 27017)
    self.db = conn["kaoqin"]
    tornado.web.Application.__init__(self, handlers, **settings)


class MainHandler(tornado.web.RequestHandler):
  def get(self):
    pass

class ReadHandler(tornado.web.RequestHandler):
  def get(self):
    #獲取collection
    coll_record = self.application.db.record
    coll_user = self.application.db.user
    #讀取excel表格
    table = xlrd.open_workbook('/Users/ant/Webdev/python/excel/data.xls')
    #讀取打卡記錄sheet
    sheet=table.sheet_by_index(2)
    #讀取打卡月份范圍
    row3 = sheet.row_values(2)
    m1 = int(row3[2][5:7])
    m2 = int(row3[2][18:20])
    #設(shè)置當(dāng)前年份
    y = int(row3[2][0:4])
    #設(shè)置當(dāng)前月份為第一個(gè)月份
    m = m1
    #讀取打卡日期范圍
    row4 = sheet.row_values(3)
    #初始化上一天
    lastday = row4[0]
    #遍歷第四行中的日期
    for d in row4:
      #如果日期小于上一個(gè)日期
      #說明月份增大，則修改當(dāng)前月份為第二個(gè)月份
      if d < lastday:
        m = m2
        #如果當(dāng)前兩個(gè)月份分別為12月和1月
        #說明跨年了，所以年份 +1
        if m1 == 12 and m2 == 1:
          y = y + 1
      #用n計(jì)數(shù)，范圍為 3 到（總行數(shù)/2+1）
      #（總行數(shù)/2+1）- 3 = 總用戶數(shù)
      #即遍歷所有用戶
      for n in range(3, sheet.nrows/2+1):
        #取該用戶的第一行，即用戶信息行
        row_1 = sheet.row_values(n*2-2)
        #獲取用戶id
        u_id = row_1[2]
        #獲取用戶姓名
        u_name = row_1[10]
        #獲取用戶部門
        u_dept = row_1[20]
        #查詢該用戶
        user = coll_user.find_one({"id":u_id})
        #如果數(shù)據(jù)庫中不存在該用戶則創(chuàng)建新用戶
        if not user:
          user = dict()
          user['id'] = u_id
          user['name'] = u_name
          user['dept'] = u_dept
          coll_user.insert(user)
        #取該用戶的第二行，即考勤記錄行
        row_2 = sheet.row_values(n*2-1)
        #獲取改當(dāng)前日期的下標(biāo)
        idx = row4.index(d)
        #獲取當(dāng)前用戶當(dāng)前日期的考勤記錄
        check_data = row_2[idx]
        #初始化空考勤記錄列表
        check = list()
        #5個(gè)字符一組，遍歷考勤記錄并存入考勤記錄列表
        for i in range(0,len(check_data)/5):
          check.append(check_data[i*5:i*5+5])
        #查詢當(dāng)前用戶當(dāng)天記錄
        record = coll_record.find_one({"y":y, "m":m, "d":d, "id":user['id']})
        #如果記錄存在則更新記錄
        if record:
          for item in check:
            #將新的考勤記錄添加進(jìn)之前的記錄
            if item not in record['check']:
              record['check'].append(item)
              coll_record.save(record)
        #如果記錄不存在則插入新紀(jì)錄
        else:
          record = {"y":y, "m":m, "d":d, "id":user['id'], "check":check}
          coll_record.insert(record)

class DayHandler(tornado.web.RequestHandler):
  def get(self):
    #獲取年月日參數(shù)
    y = self.get_argument("y",None)
    m = self.get_argument("m",None)
    d = self.get_argument("d",None)
    #判斷參數(shù)是否設(shè)置齊全
    if y and m and d:
      #將參數(shù)轉(zhuǎn)換為整型數(shù)，方便使用
      y = int(y)
      m = int(m)
      d = int(d)
      #獲取當(dāng)天所有記錄
      coll_record = self.application.db.record
      record = coll_record.find({"y":y, "m":m, "d":d})
      #獲取當(dāng)天為星期幾
      weekday = datetime.datetime(y,m,d).strftime("%w")
      #獲取當(dāng)天值班表
      coll_duty = self.application.db.duty
      duty = coll_duty.find_one({"id":int(weekday)})
      #初始化空目標(biāo)記錄（當(dāng)天值班人員記錄）
      target = list()
      #遍歷當(dāng)天所有記錄
      for item in record:
        #當(dāng)該記錄的用戶當(dāng)天有值班任務(wù)時(shí)，計(jì)算并存入target數(shù)組
        if int(item['id']) in duty['list']:
          #通過用戶id獲取該用戶值班起止時(shí)間
          start = duty[item['id']][0]
          end = duty[item['id']][1]
          #計(jì)算值班時(shí)長/秒
          date1 = datetime.datetime(y,m,d,int(start[:2]),int(start[-2:]))
          date2 = datetime.datetime(y,m,d,int(end[:2]),int(end[-2:]))
          item['length'] = (date2 - date1).seconds
          #初始化實(shí)際值班百分比
          item['per'] = 0
          #初始化上下班打卡時(shí)間
          item['start'] = 0
          item['end'] = 0
          #遍歷該用戶打卡記錄
          for t in item['check']:
            #當(dāng)比值班時(shí)間來得早
            if t < start:
              #計(jì)算時(shí)間差
              date1 = datetime.datetime(y,m,d,int(start[:2]),int(start[-2:]))
              date2 = datetime.datetime(y,m,d,int(t[:2]),int(t[-2:]))
              dif = (date1 - date2).seconds
              #當(dāng)打卡時(shí)間在值班時(shí)間前半小時(shí)內(nèi)
              if dif <= 1800:
                #上班打卡成功
                item['start'] = start
            elif t < end:
              #如果還沒上班打卡
              if not item['start']:
                #則記錄當(dāng)前時(shí)間為上班打卡時(shí)間
                item['start'] = t
              else:
                #否則記錄當(dāng)前時(shí)間為下班打卡時(shí)間
                item['end'] = t
            else:
              #如果已經(jīng)上班打卡
              if item['start']:
                #計(jì)算時(shí)間差
                date1 = datetime.datetime(y,m,d,int(end[:2]),int(end[-2:]))
                date2 = datetime.datetime(y,m,d,int(t[:2]),int(t[-2:]))
                dif = (date1 - date2).seconds
                #當(dāng)打卡時(shí)間在值班時(shí)間后半小時(shí)內(nèi)
                if dif <= 1800:
                  #下班打卡成功
                  item['end'] = end
          #當(dāng)上班下班均打卡
          if item['start'] and item['end']:
            #計(jì)算實(shí)際值班時(shí)長
            date1 = datetime.datetime(y,m,d,int(item['start'][:2]),int(item['start'][-2:]))
            date2 = datetime.datetime(y,m,d,int(item['end'][:2]),int(item['end'][-2:]))
            dif = (date2 - date1).seconds
            #計(jì)算（實(shí)際值班時(shí)長/值班時(shí)長）百分比
            item['per'] = int(dif/float(item['length']) * 100)
          else:
            #未正常上下班則視為未值班
            item['start'] = 0
            item['end'] = 0
          #將記錄添加到target數(shù)組中
          target.append(item)
      #輸出數(shù)據(jù)
      self.render("index.html",
        target = target
        )


def main():
  tornado.options.parse_command_line()
  http_server = tornado.httpserver.HTTPServer(Application())
  http_server.listen(options.port)
  tornado.ioloop.IOLoop.instance().start()


if __name__ == "__main__":
  main()

  index.html

{
{% for item in target %}
  {
   'id':{{ item['id'] }},
   'start':{{ item['start'] }},
   'end':{{ item['end'] }},
   'length':{{ item['length'] }}, 
   'per':{{ item['per'] }}
   }
{% end %}
}