import requests
session = requests.Session()  # 会话对象让你能够跨请求保持某些参数,它也会在同一个Session实例发出的所有请求之间保持cookie
session.headers = {
  'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.84 Safari/537.36',
  'Cookie': 'ASP.NET_SessionId=1qri0rmoylpyrs45rurzme55; Hm_lvt_ed06d5e5f94d85932b82e4aac94d0c68=1467535679,1469713840; Hm_lpvt_ed06d5e5f94d85932b82e4aac94d0c68=1469713840; PHPSESSID=ev339udv0rrhqg6tfdvfukqos1'
}

def get_mysql():
  conn = pymysql.connect(host = 'localhost', user = 'root', passwd = '2014081029', db = 'mysql', charset = 'utf8')  # user为数据库的名字，passwd为数据库的密码，一般把要把字符集定义为utf8，不然存入数据库容易遇到编码问题
  cur = conn.cursor()  # 获取操作游标
  cur.execute('use book')  # 使用book这个数据库
  return (cur, conn)

def get_book_name(book_url):
  html = session.get(book_url, cookies = cookie, headers = headers).content.decode('utf-8')
  soup = BeautifulSoup(html, 'lxml')
  book_bar = []  # 书籍的条形码列表,用来判断要存入数据库的书籍是否已经存在

  cur, conn = get_mysql()
  sql = 'select * from book_list;'
  cur.execute(sql)
  rows = cur.fetchall()
  for row in rows:
    book_bar.append(row[1])

  book_list = []  # 这个是我测试时使用的，作用是把每本书籍的信息列表放在这个列表中
  book_every = [] # 一本书籍的所有信息列表
  for book_time in soup.find_all('td', class_="whitetext"):
    print(book_time.get_text().strip()) # 移除字符串头尾指定的字符（默认为空格）
    pattern = re.compile(r'\s')
    content = re.sub(pattern, r'', book_time.get_text()) # 目的也是匹配任何空白符并去除，貌似对空行去除没影响

    if content != '':
      book_every.append(content)
      if len(book_every) == 7:
        book_list.append(book_every)
        if book_every[0] not in book_bar:
          sql = 'insert book_list(条形码, 题名和作者, 借阅日期, 应还日期, 续借量, 馆藏地, 附件) value(' + "\'" \
             + book_every[0] + "\'," + "\'" + book_every[1] + "\'," + "\'" + book_every[2] + "\'," + "\'" \
             + book_every[3] + "\'," + "\'" + book_every[4] + "\'," + "\'" + book_every[5] + "\'," + "\'" \
             + book_every[6] + "\'" + ');'
        try:

          cur.execute(sql)
          conn.commit()
        except:
          conn.rollback()
        book_every = []

  print(book_list)

if len(book_every) == 7:
  book_list.append(book_every)
  if book_every[0] not in book_title:
    sql = 'insert book_list(条形码, 题名和作者, 借阅日期, 应还日期, 续借量, 馆藏地, 附件) value(' + "\'" \
             + book_every[0] + "\'," + "\'" + book_every[1] + "\'," + "\'" + book_every[2] + "\'," + "\'" \
             + book_every[3] + "\'," + "\'" + book_every[4] + "\'," + "\'" + book_every[5] + "\'," + "\'" \
             + book_every[6] + "\'" + ');'
    try:
      cur.execute(sql)
      conn.commit()    
    except:
      conn.rollback()  # 如果存入数据库失败，执行回滚操作
  book_every = []  

def send_message():
  day_num = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
  day_num1 = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
  sql = 'select * from book_list;'
  cur, conn = get_mysql()
  cur.execute(sql)
  rows = cur.fetchall()
  local_time = time.strftime("%Y-%m-%d", time.localtime()) # 获取当前时间
  local_time = str(local_time)
  times = re.split(r'-', local_time)
  year = times[0]

  number = 0
  while(True):
    for i in rows:
      print(i[4])
      pattern = re.split(r'-', i[4])
      if times[1] == pattern[1]:
        day = int(times[2]) - int(pattern[2])
        if day > 0:
          print('已经超期了%d天' % day)
          number += 1
          send_email(day, number, i[2])
      elif times[1] > pattern[1]:
        if (year % 4 == 0 and year % 100 != 0) or year % 400 == 0:
          extend_day = day_num1[int(pattern[1]) - 1] - int(pattern[2]) + times[2]
          print('已经超期了%d天' % extend_day)
          number += 1
          send_email(day, number, i[2])
        else:
          extend_day = day_num[int(pattern[1]) - 1] - int(pattern[2]) + times[2]
          print('已经超期了%d天' % extend_day)
          number += 1
          send_email(day, number, i[2])

      else:
        print('还没有超期的书籍')

      print(pattern[2])
    time.sleep(3600 * 24)

      if times[1] == pattern[1]:
        day = int(times[2]) - int(pattern[2])
        if day > 0:
          print('已经超期了%d天' % day)
          number += 1
          send_email(day, number, i[2])

      elif times[1] > pattern[1]:
        if (year % 4 == 0 and year % 100 != 0) or year % 400 == 0:
          extend_day = day_num1[int(pattern[1]) - 1] - int(pattern[2]) + times[2]
          print('已经超期了%d天' % extend_day)
          number += 1
          send_email(day, number, i[2])

def send_email(day, number, title):
  from_addr = '15602200534@163.com'
  password = '就不告诉你'
  to_addr = '673411814@qq.com'
  smtp_server = 'smtp.163.com'

  text = 'Hello ,郭伟匡, 告诉你一个不好的消息,赶紧带上你的书,去图书馆交钱吧！你有一本叫《%s》的书籍超期了' \
      ',而且已经超期了%d天了,总共有%d书超期了！！！' % (title, day, number)
  msg = MIMEText(text, 'plain', 'utf-8')
  msg['From'] = format_addr('图书馆的通知<%s>' % from_addr)
  msg['To'] = format_addr('管理员<%s>' % to_addr)
  msg['Subject'] = Header('来着郭伟匡的问候......', 'utf-8').encode()

  server = smtplib.SMTP(smtp_server, 25)
  server.set_debuglevel(1)
  server.login(from_addr, password)
  server.sendmail(from_addr, [to_addr], msg.as_string())
  server.quit()