Python Busclock

Author: alisx

taiyangxue/busclock/BusClock.py

@@ -0,0 +1,118 @@
+import httpx
+import mail
+import datetime
+from bs4 import BeautifulSoup as bs4
+import re
+import config as conf
+import os
+
+class BjBusClock():
+    def __init__(self, logger, config=None, ):
+        self.logger = logger
+        import json
+        if config is None:
+            with open(r"C:\config.json", "r", encoding='UTF-8') as config_file:
+                config = json.load(config_file)
+        if config is None:
+            raise "没有提供配置且没找到配置文件"
+
+        self.config = conf.Config(config)  # 整体配置
+        
+        if self.config.mailConfig is None:
+            raise "没有发送通知的邮箱配置"
+        self.config.alertMail = config.get("alertMail", None)
+        if self.config.alertMail is None:
+            raise "没有接收通知的邮箱地址"
+        self.mailClient = mail.MailSent(self.config.mailConfig)
+        self.logger.info("BusClock 初始化完毕")
+
+
+    def run(self):
+        self.logger.info("BusClock 执行运算")
+        for line in self.config.lines:
+            self.logger.info("BusClock 执行线路:" + "@" + line['line'])
+            if self.onTime(line['begin'], line['end']):
+                self.logger.info("BusClock 达到提示时间 " + "@" + line['line'])
+                if line.get('needSentMail', True):
+                    self.logger.info("BusClock 需要发提醒 " + "@" + line['line'])
+                    bustime = self.getBusTime(line)
+                    if  bustime is None:
+                        self.logger.info("BusClock 尚未发车: " + "@" + line['line'])
+                    else:
+                        self.logger.info("BusClock 得到车辆时间: " + bustime + "@" + line['line'])
+                        if int(bustime) <= int(line.get('latestLeaveMinute', self.config.latestLeaveMinute)):
+                            self.logger.info("BusClock 可以发邮件提醒了" + "@" + line['line'])
+                            self.mailClient.send_mail(self.config.alertMail, '班车提醒: '+line['line'], '车辆即将到站，现在出发正当时')
+                            self.logger.info("BusClock 邮件提醒已发送" + "@" + line['line'])
+                            line['needSentMail'] = False  # 发送通知后，不必再发了
+                        else:
+                            self.logger.info("BusClock 没到发邮件的时候" + "@" + line['line'])
+                else:
+                    self.logger.info("BusClock 已经发过邮件了" + "@" + line['line'])
+            else:
+                line['needSentMail'] = True
+                self.logger.info("BusClock 未到时间窗口" + "@" + line['line'])
+
+
+    def onTime(self, begin, end):
+        d_time = datetime.datetime.strptime(
+            str(datetime.datetime.now().date())+begin, '%Y-%m-%d%H:%M')
+        d_time1 = datetime.datetime.strptime(
+            str(datetime.datetime.now().date())+end, '%Y-%m-%d%H:%M')
+        n_time = datetime.datetime.now()
+        if n_time > d_time and n_time < d_time1:
+            return True
+        else:
+            return False
+
+    def getBusTime(self, line):
+        url = 'http://www.bjbus.com/home/ajax_rtbus_data.php?act=busTime&selBLine=%(line)s&selBDir=%(dir)s&selBStop=%(stop)s' % line
+        result = None
+        try:
+            r = httpx.get(url).json()
+            b = bs4(r.get('html'), 'html.parser')
+            info = b.find('article')
+            i = info.find_all('p')[1]
+            ret = re.search(r'\d+(?=\s分钟)', i.text)
+            result = None
+            if ret:
+                result = ret.group()
+        except BaseException:
+            pass
+
+        return result
+
+
+if __name__ == "__main__":
+    import logging
+    logging.basicConfig(filename=r'D:\busclock.log', level=logging.INFO)
+    bus = BjBusClock(logging, {
+        "loopWaitSeconds": 60,
+        "spurtWaitSeconds": 10,
+        "latestLeaveMinute": 5,
+        "mailConfig": {
+            "FROM": "tom@example.com",
+            "HOST": "smtp.example.com",
+            "PORT": "465",
+            "USER": "tom",
+            "PASS": "password",
+            "SSL": True
+         },
+        "alertMail": "lily@example.com",
+        "lines": [{
+            "line": "835快",
+            "dir": "5066222788346588777",
+            "stop": "13",
+            "begin": "08:00",
+            "end": "08:30"
+        }, {
+            "line": "835快",
+            "dir": "4997908670784162973",
+            "stop": "3",
+            "begin": "19:00",
+            "end": "20:30"
+        }]
+    })
+    print(bus.onTime("12:00", "13:00"))
+    print(bus.getBusTime(bus.config.lines[0]))
+    print(bus.run())

taiyangxue/busclock/config.json

@@ -0,0 +1,27 @@
+{
+    "loopWaitSeconds": 60,
+    "spurtWaitSeconds": 10,
+    "latestLeaveMinute": 5,
+    "mailConfig": {
+      "FROM": "tom@example.com",
+      "HOST": "smtp.example.com",
+      "PORT": "465",
+      "USER": "tom",
+      "PASS": "password",
+      "SSL": true
+    },
+    "alertMail": "lily@example.com",
+    "lines": [{
+      "line": "835快",
+      "dir": "5066222788346588777",
+      "stop": "13",
+      "begin": "08:00",
+      "end": "08:30"
+    }, {
+      "line": "835快",
+      "dir": "4997908670784162973",
+      "stop": "3",
+      "begin": "19:00",
+      "end": "20:30"
+    }]
+  }

taiyangxue/busclock/config.py

@@ -0,0 +1,7 @@
+class Config:
+    def __init__(self, config):
+        self.loopWaitSeconds = config.get("loopWaitSeconds", 60)
+        self.spurtWaitSeconds = config.get("spurtWaitSeconds", 10)
+        self.mailConfig = config.get("mailConfig", None)
+        self.latestLeaveMinute = config.get("latestLeaveTime", 5)
+        self.lines = config.get("lines", {})

taiyangxue/busclock/mail.py

@@ -0,0 +1,53 @@
+#!/usr/local/bin/python3
+# coding=utf-8
+
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+
+
+class MailSent(object):
+    default_config = {
+        'PORT': '465',
+        'SSL': True
+    }
+
+    def __init__(self, config):
+        self.config = {
+            'FROM': config.get('mail_from', self.default_config['FROM']),
+            'HOST': config.get('mail_host', self.default_config['HOST']),
+            'PORT': config.get('port', self.default_config['PORT']),
+            'USER': config.get('user', self.default_config['USER']),
+            'PASS': config.get('mail_pass', self.default_config['PASS']),
+            'SSL': config.get('ssl', self.default_config['SSL']),
+        }
+
+    def send_mail(self, to_mail, title, content):
+        ret = True
+        FROM_MAIL = self.config['FROM']  # 发件人
+        if type(to_mail) == 'list':
+            TO_MAIL = to_mail  # 收件人
+        else:
+            TO_MAIL = [to_mail]
+
+        SMTP_SERVER = self.config['HOST']  # qq邮箱服务器
+        SSL_PORT = self.config['PORT']  # 加密端口
+        USER_NAME = self.config['USER']  # qq邮箱用户名
+        USER_PWD = self.config['PASS']  # qq邮箱授权码
+        msg = MIMEMultipart('alternative')  # 实例化email对象
+        msg['from'] = FROM_MAIL  # 对应发件人邮箱昵称、发件人邮箱账号
+        msg['to'] = ';'.join(TO_MAIL)  # 对应收件人邮箱昵称、收件人邮箱账号
+        msg['subject'] = title  # 邮件的主题
+        txt = MIMEText(content, 'html')
+        msg.attach(txt)
+        try:
+            # 纯粹的ssl加密方式
+            smtp = smtplib.SMTP_SSL(SMTP_SERVER, SSL_PORT)  # 邮件服务器地址和端口
+            smtp.ehlo()  # 用户认证
+            smtp.login(USER_NAME, USER_PWD)  # 括号中对应的是发件人邮箱账号、邮箱密码
+            smtp.sendmail(FROM_MAIL, TO_MAIL, str(msg))  # 收件人邮箱账号、发送邮件
+            smtp.quit()  # 等同 smtp.close()  ,关闭连接
+            # print('发送成功')
+        except Exception:
+            ret = False
+        return ret

taiyangxue/busclock/requirements.txt

@@ -0,0 +1,68 @@
+import sys
+import time
+from datetime import datetime
+
+import win32api
+import win32event
+import win32service
+import win32serviceutil
+import servicemanager
+import logging
+
+import BusClock
+
+
+class MyService(win32serviceutil.ServiceFramework):
+
+    _svc_name_ = "MYBusClock"
+    _svc_display_name_ = "MyBusClock"
+    _svc_description_ = "提醒公交车到站时间，帮助我节约时间"
+
+    def __init__(self, args):
+        logging.basicConfig(filename=r'D:\busclock.log', level=logging.INFO)
+        logging.info("init")
+        win32serviceutil.ServiceFramework.__init__(self, args)
+        self.stop_event = win32event.CreateEvent(None, 0, 0, None)
+        self.isAlive = True
+        self.loopWaitSeconds = 5
+        self.busClock = BusClock.BjBusClock(logging)
+
+    def SvcDoRun(self):
+        self.ReportServiceStatus(win32service.SERVICE_START_PENDING)
+        try:
+            self.ReportServiceStatus(win32service.SERVICE_RUNNING)
+            logging.info('start')
+            self.start()
+            logging.info('wait')
+            win32event.WaitForSingleObject(
+                self.stop_event, win32event.INFINITE)
+            logging.info('done')
+        except BaseException as e:
+            logging.info('Exception : %s' % e)
+            self.SvcStop()
+
+    def SvcStop(self):
+        self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
+        logging.info('stopping')
+        self.stop()
+        logging.info('stopped')
+        win32event.SetEvent(self.stop_event)
+        self.ReportServiceStatus(win32service.SERVICE_STOPPED)
+
+    def start(self):
+        logging.info("svc do run....")
+        while self.isAlive:
+            self.busClock.run()
+            time.sleep(self.loopWaitSeconds)
+
+    def stop(self):
+        self.isAlive = False
+        pass
+
+if __name__ == "__main__":
+    if len(sys.argv) == 1:
+        servicemanager.Initialize()
+        servicemanager.PrepareToHostSingle(MyService)
+        servicemanager.StartServiceCtrlDispatcher()
+    else:
+        win32serviceutil.HandleCommandLine(MyService)