ホームページ > 記事 > バックエンド開発 > Pythonリクエストでログインをシミュレートし、図書館の座席の自動予約を実現します

Pythonリクエストでログインをシミュレートし、図書館の座席の自動予約を実現します

不言オリジナル: 2018-04-27 10:40:524652ブラウズ

この記事では主に Python リクエストの模擬ログインについて詳しく紹介します。Python は図書館の座席の自動予約を実現します。興味のある方は参考にしてください。

この記事の例は Python の実装本を共有しています。ホールの座席を自動予約するためのコードは参考用です。具体的な内容は次のとおりです。

設定

パブリックネットワークホスト経由でスクリプトを定期的に実行し、QQ メールボックスに電子メールを送信します。予約が成功したかどうかを確認する WeChat のメッセージ

vim /etc/crontab

毎朝 7:01 ごとに自動的に実行されるようにスクリプトを設定します

プログラムの流れ

(yuyue を取ってください。 juneberry.cn Web サイトを例として)

ログインページにアクセスし、Cookie とフォーム内の非表示の投稿フィールドを取得します
でログイン投稿データを構築し、非表示の投稿フィールドを追加します
投稿が構築されましたフォームからデータを取得し、ログインをシミュレートし、Cookie をアクティブ化します (Cookie にログイン権限を持たせます)
座席予約インターフェースへのアクセスを取得し、Cookie を有効化します (Cookie に座席予約権限を持たせます)
予約リクエストを送信します。座席予約を実現します
返された結果を解析し、成功したかどうかを判断し、メールリマインダーを送信します

Points

リクエストライブラリのrequests.session() Cookie を渡すことができるセッション
c726a8da2e1b0a50980b1ad4157a77d3データを取得し、それをポストに渡します
データ内のパケットをキャプチャしてウェブサイトのロジックを決定し、各リクエストのパラメーターをフィルターで除外します。そして、プログラム_get_date_str(self):
に

関数の説明

class FUCK()main class
を実装します_get_date_str(self):この関数を使用して、現在の日付を取得し、1日を追加しますURL の特性フィールドを構築します (ライブラリは 1 日前に座席を予約するように設定されています)
def _get_order_url(self):「座席を予約」投稿ターゲット URL を構築します
def _get_static_post_attr: この関数は、get リクエストの戻りページを解析し、後続の投稿データの構築のために c726a8da2e1b0a50980b1ad4157a77d3 のフィールドを抽出します
def login(self):ログイン関数を実現します
def run(self):座席予約機能を実現
def _is_success(self, text):予約結果を判定
def error_log_once( self, text='default error (once) '):
def error_log(self, text='default error'):これらの 2 つの関数は、プログラムのステータスを「間違っている」または「間違っていません」に設定します。ステータス (繰り返しの書き込みを避けるために使用されます)自動化実行時のエラー情報をログに記録します)
def error_log(self, text='default error'):エラー情報をローカルログに一度書き込みます
sendmail.send_mail()メール送信モジュール

コードとコメント

# /bin/python
# -*- coding:utf-8 -*-
import time
import sys
import requests
from bs4 import BeautifulSoup
from mail import sendmail

__author__ = &#39;xy&#39;

# 主类
class FUCK():
 def __init__(self, username, password, seatNO, mailto):
 """
  以四个参数初始化，用户名，密码，要预约的座位号，接受预约结果提醒邮件的邮箱
 """
  self.username = username
  self.password = password
  self.seatNO = seatNO
  self.mailto = mailto
  self.base_url = &#39;http://yuyue.juneberry.cn&#39;
  self.login_url = &#39;http://yuyue.juneberry.cn&#39;
  self.order_url = self._get_order_url()

  self.login_content = &#39;&#39;
  self.middle_content = &#39;&#39;
  self.final_content = &#39;&#39;

  self.s = requests.session() # 创建可传递cookies的会话

  # post data for login
  self.data1 = {
   &#39;subCmd&#39;: &#39;Login&#39;,
   &#39;txt_LoginID&#39;: self.username, # S+学号
   &#39;txt_Password&#39;: self.password, # 密码
   &#39;selSchool&#39;: 60, # 60表示北京交通大学
  }

  # post data for order a seat
  self.data2 = {
   &#39;subCmd&#39;: &#39;query&#39;,
  }

  # 自定义http头，然而我在程序里并没有使用
  self.headers = {
   &#39;Connection&#39;: &#39;keep-alive&#39;,
   &#39;Content-Type&#39;: &#39;application/x-www-form-urlencoded&#39;,
  }

  self.login()
  self.run()
  self._is_success(self.final_content)

  # 怀疑程序出错时，取消下行注释，可打印一些参数
  # self._debug()

 def _get_date_str(self):
  s = time.localtime(time.time())
  ########333
  date_str = str(s.tm_year) + &#39;%2f&#39; + str(s.tm_mon) + &#39;%2f&#39; + str(s.tm_mday + 1)
  date_str = date_str.replace(&#39;%2f1%2f32&#39;, &#39;%2f2%2f1&#39;) \
   .replace(&#39;%2f2%2f29&#39;, &#39;%2f3%2f1&#39;) \
   .replace(&#39;%2f3%2f32&#39;, &#39;%2f4%2f1&#39;) \
   .replace(&#39;%2f4%2f31&#39;, &#39;%2f5%2f1&#39;) \
   .replace(&#39;%2f5%2f32&#39;, &#39;%2f6%2f1&#39;) \
   .replace(&#39;%2f6%2f31&#39;, &#39;%2f7%2f1&#39;) \
   .replace(&#39;%2f7%2f32&#39;, &#39;%2f8%2f1&#39;) \
   .replace(&#39;%2f8%2f32&#39;, &#39;%2f9%2f1&#39;) \
   .replace(&#39;%2f9%2f31&#39;, &#39;%2f10%2f1&#39;) \
   .replace(&#39;%2f10%2f32&#39;, &#39;%2f11%2f1&#39;) \
   .replace(&#39;%2f11%2f31&#39;, &#39;%2f12%2f1&#39;) \
   .replace(&#39;%2f12%2f32&#39;, &#39;%2f1%2f1&#39;)
  return date_str

 def _get_order_url(self):
  return "http://yuyue.juneberry.cn/BookSeat/BookSeatMessage.aspx?seatNo=101001" + self.seatNO + "&seatShortNo=01" + self.seatNO + "&roomNo=101001&date=" + self._get_date_str()

 def _get_static_post_attr(self, page_content, data_dict):
  """
  拿到<input type=&#39;hidden&#39;>的post参数，并添加到post_data中
  """
  soup = BeautifulSoup(page_content, "html.parser")
  for each in soup.find_all(&#39;input&#39;):
   if &#39;value&#39; in each.attrs and &#39;name&#39; in each.attrs:
    data_dict[each[&#39;name&#39;]] = each[&#39;value&#39;] # 添加到login的post_data中
    # self.data2[each[&#39;name&#39;]] = each[&#39;value&#39;] # 添加到order的post_data中
  return data_dict

 def _debug(self):

  print self.order_url
  print self.data1
  print self.data2
  print self.headers
  print self.s.cookies

  # print self.login_content
  # print self.middle_content
  print self.final_content

 def login(self):
  homepage_content = self.s.get(self.base_url).content
  self.data1 = self._get_static_post_attr(homepage_content, self.data1)
  r = self.s.post(self.login_url, self.data1)
  self.login_content = r.content

 def run(self):

  # 这个get的意思是：原先的cookie没有预约权限，
  # 访问这个get之后，会使cookie拥有预约权限，从而执行下一个post
  self.middle_content = self.s.get(&#39;http://yuyue.juneberry.cn/BookSeat/BookSeatListForm.aspx&#39;).content

  # 经测试，这个post只需要一个subCmd的参数就可以正常返回，因此不必根据get内容更新post参数
  # self.data2 = self._get_static_post_attr(middle_content, self.data2)

  # 这个post请求完成了预约功能！
  r = self.s.post(self.order_url, self.data2)

  self.final_content = r.content

 def _is_success(self, text):
  """
  接受最终的html内容，判断是否成功，并触发日志记录和邮件提醒
  """
  if &#39;<h5 id="MessageTip">已经存在有效的预约记录。</h5>&#39; in text:
   self.clear_error_once(&#39;[done!] You already ordered a seat!&#39;)
  elif &#39;<h5 id="MessageTip">选择的日期不允许预约。</h5>&#39; in text:
   self.clear_error_once(&#39;[done!] Date is wrong!&#39;)
  elif &#39;<h5 id="MessageTip">所选座位已经被预约。</h5>&#39; in text:
   self.clear_error_once(&#39;[done!] This seat is not available, maybe taken by others!&#39;)
  elif &#39;<h5 id="MessageTip">座位预约成功&#39; in text:
   self.clear_error_once(&#39;[done!] Success! An email is sending to you!&#39;)
   sendmail.send_mail(&#39;BJTU Library Seat_NO:&#39; + self.seatNO + &#39;ordered!&#39;,
        &#39;Sending by robot. Do not reply this mail!&#39;, self.mailto)
  else:
   self.error_log_once(&#39;Error! 302 to login page&#39;)

 def error_log_once(self, text=&#39;default error (once)&#39;):
  try:
   is_error_file = open(&#39;./isopen_xy.txt&#39;, &#39;r&#39;)
  except:
   is_error_file = open(&#39;./isopen_xy.txt&#39;, &#39;w&#39;)
  if &#39;1&#39; not in is_error_file.read():
   print &#39;writting error to log...&#39;
   self.error_log(text)
  else:
   print &#39;already written to log&#39;
  is_error_file.close()
  sendmail.send_mail(&#39;BJTU_Library system error once !&#39;, &#39;error text!&#39;)

 def error_log(self, text=&#39;default error&#39;):
  is_error_file = open(&#39;./isopen_xy.txt&#39;, &#39;w&#39;)
  is_error_file.write(&#39;1\n&#39;)
  is_error_file.close()

  f = open("./log_xy.txt", &#39;a&#39;)
  f.write(time.strftime("%Y-%m-%d %X", time.localtime()) + text + &#39;\n&#39;)
  f.close()

 def clear_error_once(self, text=&#39;success&#39;):
  print text
  is_error_file = open(&#39;./isopen_xy.txt&#39;, &#39;w&#39;)
  is_error_file.write(&#39;0\n&#39;)
  is_error_file.close()


if __name__ == &#39;__main__&#39;:
 if len(sys.argv) < 5:
  print &#39;Usage: python library.py [username] [password] [seat_NO] [email]&#39;
  print &#39;eg. python library.py S13280001 123456 003 XXXX@qq.com\n&#39;
  print &#39;Any problems, mail to: i[at]cdxy.me&#39;
  print &#39;#-*- Edit by cdxy 16.03.24 -*-&#39;
  sys.exit(0)
 else:
  FUCK(sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4])

Pythonリクエストでログインをシミュレートし、図書館の座席の自動予約を実現します

関連記事