电影代码升级，输入任何类型的电影都能看到

import requests import re import xlwt import os def main(): d=input("请输入（爱情片，动作片，喜剧片，战争片，科幻片，剧情片，武侠片，冒险片，枪战片，恐怖片）：") if d=="爱情片": e="aiqingpian" xieRu(e) elif d=="动作片": e="dongzuopian" xieRu(e) elif d=="喜剧片": e="xijupian" xieRu(e) elif d=="战争片": e="zhanzhengpian" xieRu(e) elif d=="科幻片": e="kehuanpian" xieRu(e) elif d=="剧情片": e="juqingpian" xieRu(e) elif d=="武侠片": e="wuxiapian" xieRu(e) elif d=="冒险片": e="maoxianpian" xieRu(e) elif d=="枪战片": e="qiangzhanpian" xieRu(e) elif d=="恐怖片": e="kongbupian" xieRu(e) else: print("您的输入不正确，请重新输入") main() def xieRu(e): workbook = xlwt.Workbook(encoding="utf-8") worksheet = workbook.add_sheet('My Worksheet') biao_tou = ["电影名称", "电影网址(复制链接到浏览器可直接观看)"] for g in range(len(biao_tou)): worksheet.write(0, g, biao_tou[g]) if os.path.exists("{}.xls".format(e))==False: workbook.save("{}.xls".format(e)) else: os.remove("{}.xls".format(e)) workbook.save("{}.xls".format(e)) K = int(input("请输入页数:")) h = 0 t = 0 for i in range(1, K + 1): url = "https://www.4480qpg.com/show/{}/page/{}.html".format(e, i) response = requests.get(url) a = re.findall(".class=\"stui-vodlist__thumb lazyload\" href=\"(.*?)\".", response.text) b = re.findall(".class=\"stui-vodlist__thumb lazyload\" href=\"/vod/\d{0,10}.html\" title=\"(.*?)\".", response.text) for j in range(len(a)): url_d = "https://www.4480qpg.com{}".format(a[j]) worksheet.write(h + 1, 1, url_d) if os.path.exists("{}.xls".format(e)) == False: workbook.save("{}.xls".format(e)) else: os.remove("{}.xls".format(e)) workbook.save("{}.xls".format(e)) h = h + 1 print(url_d) for g in range(len(b)): worksheet.write(t + 1, 0, b[g]) if os.path.exists("{}.xls".format(e)) == False: workbook.save("{}.xls".format(e)) else: os.remove("{}.xls".format(e)) workbook.save("{}.xls".format(e)) t = t + 1 print(b[g]) if __name__ == '__main__': main()