python, 實現朋友家人年曆生日自動提醒

爲了方便提醒本身，今天有哪位朋友過生日（年曆生日）。

測試環境：fedora25桌面版。

 

創建一個生日配置文件，注意：按日期排好序

vi /etc/birthday.txt

朋友A 1-4
朋友C 2-3
朋友B 3-8

 

建立腳本文件

建立腳本文件 /usr/bin/check_birthday， 腳本內容以下。

#!/usr/bin/python
# -*- coding: utf-8 -*-

from lunardate import LunarDate
import os
import sys

def get_month_day(datestring):
    array = datestring.split("-")

    if len(array) == 3:
        (y, m, d) = array
        return int(m), int(d)
    else:
        (m, d) = array
        return int(m), int(d)


def is_birthday_today(date, birthday):
    m, d = get_month_day(birthday)
    return m == date.month and d == date.day


def distance_birthday_today(date, birthday):
    m, d = get_month_day(birthday)
    return (m - date.month) * 30 + d - date.day


def showtip(line):
    os.system('zenity --info --title "title" --text "' + line + '" --width=300 --height=200')


filename='/etc/birthday.txt'
lines = open(filename).readlines()

lines = map(lambda x: x.strip(), lines)
lines = filter(lambda x: x != "", lines)


if len(lines) == 0 or lines[0].strip() == "":
  print("配置文件%s中沒有數據" % filename)
  sys.exit(1)

today = LunarDate.today()

exist_someone_birthday = False

# 查找誰今天過生日
for line in lines:
    array = line.split()
    if is_birthday_today(today, array[1]):
        exist_someone_birthday = True
        showtip("今天生日, " + line)
        sys.exit(0)

all_finish_this_year = True
# 提醒誰快要過生日
if not exist_someone_birthday:
    for line in lines:
        array = line.split()
        if distance_birthday_today(today, array[1]) > 0:
	    all_finish_this_year = False
            showtip("今天%d-%d, %s" % (today.month, today.day, line))
            sys.exit(0)

if all_finish_this_year:
  showtip("今天%d-%d, %s" % (today.month, today.day, lines[0]))

 

設置腳本執行權限

chmod 777 /usr/bin/check_birthday

 

下載LunarDate模塊

pip install LunarDate

提示：若是系統中沒有安裝pip，先安裝pip（yum install -y pip）

 

檢查今天是否有人過生日

check_birthday

 

能夠設置天天定時檢查

經過crontab -e進行設置

# 天天12,20點檢查一次

0 12,20 * * * /usr/bin/check_birthday