【Python】複数ページのリダイレクト用JavaScriptを自動生成するプログラム

<script src="https://ajax.googleapis.com/ajax/libs/jquery/3.4.1/jquery.min.js"></script>
<script type="text/javascript" language="javascript">
// <![CDATA[
  var href = window.location.href;
  var re = new RegExp(".*entry/{old_url}");
  if (re.test(href)) {
    $(document).ready(function() {
      if ($("#main-inner").children().hasClass("no-entry")) {
        var content = "<p>本記事は移転しました。</p><p>約3秒後にリダイレクトします。</p><p>リダイレクトしない場合は<a href='{new_url}'>こちら</a>をクリックしてください。</p>" ;
        $(".entry-footer").addClass("sorry_content");
        $(".entry-footer").html(content);
      }
    });
    var url = "{new_url}";
    var link = document.getElementsByTagName("link")[0];
    link.href = url;
    setTimeout("redirect()", 3000);
    function redirect(){
    location.href = url;
    }
  }
// ]]>
</script>

# -*- coding: utf-8 -*-
#=======================設定する変数ここから=======================
#output_dir = r'/Users/'
output_dir = r''
#出力ファイルを保存するディレクトリ（末尾に区切り文字を含む）
#空白ならこのスクリプトがあるディレクトリに出力する

output_name = r'URLtest.txt'
#出力ファイル名（拡張子含む）
#=======================設定する変数ここまで=======================

import os
import sys
import re
import pandas as pd

if output_dir == '':
    output_dir = os.path.dirname(os.path.abspath(__file__)) + os.sep
    #スクリプトがあるディレクトリの絶対パスを取得

script_head = '<script src="https://ajax.googleapis.com/ajax/libs/jquery/3.4.1/jquery.min.js"></script>\n'\
              '<script type="text/javascript" language="javascript">'
script = '// <![CDATA[\n'\
        '  var href = window.location.href;\n'\
        '  var re = new RegExp(".*entry/{old_url}");\n'\
        '  if (re.test(href)) {\n'\
        '    $(document).ready(function() {\n'\
        '      if ($("#main-inner").children().hasClass("no-entry")) {\n'\
        '        var content = "<p>本記事は移転しました。</p><p>約3秒後にリダイレクトします。</p><p>リダイレクトしない場合は'\
        '<a href=\'{new_url}\'>'\
        'こちら</a>をクリックしてください。</p>" ;\n'\
        '        $(".entry-footer").addClass("sorry_content");\n'\
        '        $(".entry-footer").html(content);\n'\
        '      }\n'\
        '    });\n'\
        '    var url = "{new_url}";\n'\
        '    var link = document.getElementsByTagName("link")[0];\n'\
        '    link.href = url;\n'\
        '\n'\
        '    setTimeout("redirect()", 3000);\n'\
        '    function redirect(){\n'\
        '    location.href = url;\n'\
        '    }\n'\
        '  }\n'\
        '// ]]>'

script_foot = '</script>'

inputs = sys.argv
#コマンドライン引数を取得
if len(inputs) != 2:
    print('[ERROR: 入力ファイルをコマンドライン引数で1つ指定してください]')
    quit()

url_file = inputs[1]
df = pd.read_excel(url_file, header=None, index_col=None)
#エクセルファイルをdata frameとして読み込み

url_lst = []
#旧URLと新URLの組を格納するリスト
for row in df.itertuples():
    #ループで各行を取得
    old_url = row[1]
    new_url = row[3]
    if (type(old_url) is str) and ('/entry/' in old_url) and (type(new_url) is str) and re.match('https?://', new_url):
    #URLが書かれているかを判定
        url_lst.append([old_url, new_url])

with open(output_dir + output_name, 'w') as output_file:
    print(script_head, file=output_file)
    for item in url_lst:
        old_entry = re.search('entry/(.*)', item[0]).group(1)
        script2 = script.replace('{old_url}', old_entry)
        script2 = script2.replace('{new_url}', item[1])
        print(script2, file=output_file)
    print(script_foot, file=output_file)

for row in df.itertuples():
    #ループで各行を取得
    flag = row[1]
    if flag == 1:
        old_url = row[2]
        new_url = row[4]
        if (type(old_url) is str) and ('/entry/' in old_url) and (type(new_url) is str) and re.match('https?://', new_url):
            url_lst.append([old_url, new_url])