Support downloading cover image if necessary.

README.md

@@ -1,6 +1,9 @@
 # DLsite_Renamer
 ![範例圖片](https://i.imgur.com/8EH3l16.png)
 
+### Purpose
+Rename the DLsite works and download the cover If required.
+
 ### How to use:
 1. Download the binary [release pack](https://github.com/ch010060/DLsite_Renamer/releases) and run .exe or run python script by yourself.
 2. There are two versions: 
@@ -77,15 +80,17 @@ VJ開頭 => 遊戲 作品
 1. config.json 文件使用 **UTF-8** 編碼, 請不要用 Windows 系統自帶的記事本進行編輯，推薦使用專業的**文件編輯器**，例如: [Notepad3](https://www.appinn.com/notepad3/), [Notepad++](https://notepad-plus-plus.org/), [vscode](https://code.visualstudio.com/)
 ![Notepad3](https://i.imgur.com/L73BXEZ.png)
 2. 去除標題中【】之間的內容主要用於去掉不必要的標註，可多加利用~
+3. 若封面(cover.jpg)已存在資料夾中，則會自動跳過下載
 
 ### (Optional) CLI version without GUI and loop：
 ```
-usage: dlsite_renamer-cli.py [-h] [-d] -i PATH
+usage: dlsite_renamer-cli.py [-h] [-d] [-c] -i PATH
 
-Renamer for DLsite works
+Renamer for DLsite works v3.0
 
 optional arguments:
  -h, --help show this help message and exit
  -d, --DEL delete string in 【】
+ -c, --COVER download cover
  -i PATH, --PATH PATH path for processing
 ```

dlsite_renamer-cli.py

@@ -11,6 +11,7 @@
 import random
 import sys
 import argparse
+import urllib.request
 
 # 默認設定
 template_RJ = 'workno title ' # 默認RJ命名模板(Voice)
@@ -150,15 +151,16 @@ def match_code(work_code):
  url = VJ_G_WEBPATH + work_code
  r = s.get(url, allow_redirects=False, cookies=R_COOKIE)
  if r.status_code != 200:
- return r.status_code, "", "", [], [], "", ""
+ return r.status_code, "", "", "", [], [], "", ""
  except os.error as err:
  print("**請求超時!\n")
  print(" 請檢查網絡連接\n")
- return "", "", "", [], [], "", ""
+ return "", "", "", "", [], [], "", ""
 
  # fromstring()在解析xml格式時, 將字串轉換為Element對像, 解析樹的根節點
  # 在python中, 對get請求返回的r.content做fromstring()處理, 可以方便進行後續的xpath()定位等
  tree = html.fromstring(r.content)
+ img_url = tree.xpath('//meta[@name="twitter:image:src"]/@content')[0]
  title = tree.xpath('//a[@itemprop="url"]/text()')[0]
  circle = tree.xpath(
  '//span[@itemprop="brand" and @class="maker_name"]/*/text()')[0]
@@ -177,14 +179,14 @@ def match_code(work_code):
  if len(release_date) >= 11:
  release_date = release_date[2]+release_date[3]+release_date[5]+release_date[6]+release_date[8]+release_date[9]
 
- return 200, title, circle, cvList, authorList, work_age[0], release_date
+ return 200, img_url, title, circle, cvList, authorList, work_age[0], release_date
 
  except os.error as err:
  print("**請求超時!\n")
  print(" 請檢查網絡連接\n")
- return "", "", "", [], [], "", ""
+ return "", "", "", "", [], [], "", ""
 
-def nameChange(path, del_flag):
+def nameChange(path, del_flag, cover_flag):
  print("選擇路徑: " + path + "\n")
  # os.listdir()返回指定的資料夾包含的檔案或資料夾的名字的列表
  files = os.listdir(path)
@@ -204,7 +206,7 @@ def nameChange(path, del_flag):
  else:
  #print('Processing: ' + code)
  print('Processing: ' + code + '\n')
- r_status, title, circle, cvList, authorList, work_age, release_date = match_code(code)
+ r_status, img_url, title, circle, cvList, authorList, work_age, release_date = match_code(code)
  # 如果順利爬取網頁訊息
  if r_status == 200 and title and circle:
  if del_flag:
@@ -239,6 +241,19 @@ def nameChange(path, del_flag):
  else:
  new_name = new_name.replace("(CV. cv)", "")
 
+
+ # 要下載封面且是資料夾
+ if cover_flag and img_url and os.path.isdir(os.path.join(path, file)):
+ try: # 嘗試下載封面
+ store_path = os.path.join(path, file, "cover.jpg")
+ if not os.path.isfile(store_path):
+ print(" 下載封面...\n")
+ urllib.request.urlretrieve(img_url, store_path)
+ else:
+ print("**封面已存在，跳過下載!\n")
+ except os.error as err:
+ print("**下載封面過程中出現錯誤!\n")
+
  # 將Windows文件名中的非法字元替換
  # re.sub(pattern, repl, string)
  new_name = re.sub(filter, " ", new_name)
@@ -277,8 +292,9 @@ def dir_path(path):
  raise argparse.ArgumentTypeError(f"\"{path}\" is not a valid path!")
 
 def process_command(): 
- parser = argparse.ArgumentParser(description="Renamer for DLsite works")
+ parser = argparse.ArgumentParser(description="Renamer for DLsite works v3.0")
  parser.add_argument('-d', "--DEL", action='store_true', help='delete string in 【】')
+ parser.add_argument('-c', "--COVER", action='store_true', help='download cover')
  parser.add_argument('-i', "--PATH", type=dir_path, required=True, help='path for processing')
  return parser.parse_args()
 
@@ -341,4 +357,4 @@ def process_command():
  print("**使用默認命名模板:\n")
  print(" workno title \n")
 
-nameChange(args.PATH,args.DEL)
+nameChange(args.PATH,args.DEL,args.COVER)

dlsite_renamer.py

@@ -9,6 +9,7 @@
 import os
 import json
 import random
+import urllib.request
 
 # 默認設定
 template_RJ = 'workno title ' # 默認RJ命名模板(Voice)
@@ -148,15 +149,16 @@ def match_code(work_code):
  url = VJ_G_WEBPATH + work_code
  r = s.get(url, allow_redirects=False, cookies=R_COOKIE)
  if r.status_code != 200:
- return r.status_code, "", "", [], [], "", ""
+ return r.status_code, "", "", "", [], [], "", ""
  except os.error as err:
  text.insert(tk.END, "**請求超時!\n")
  text.insert(tk.END, " 請檢查網絡連接\n")
- return "", "", "", [], [], "", ""
+ return "", "", "", "", [], [], "", ""
 
  # fromstring()在解析xml格式時, 將字串轉換為Element對像, 解析樹的根節點
  # 在python中, 對get請求返回的r.content做fromstring()處理, 可以方便進行後續的xpath()定位等
  tree = html.fromstring(r.content)
+ img_url = tree.xpath('//meta[@name="twitter:image:src"]/@content')[0]
  title = tree.xpath('//a[@itemprop="url"]/text()')[0]
  circle = tree.xpath(
  '//span[@itemprop="brand" and @class="maker_name"]/*/text()')[0]
@@ -175,20 +177,21 @@ def match_code(work_code):
  if len(release_date) >= 11:
  release_date = release_date[2]+release_date[3]+release_date[5]+release_date[6]+release_date[8]+release_date[9]
 
- return 200, title, circle, cvList, authorList, work_age[0], release_date
+ return 200, img_url, title, circle, cvList, authorList, work_age[0], release_date
 
  except os.error as err:
  text.insert(tk.END, "**請求超時!\n")
  text.insert(tk.END, " 請檢查網絡連接\n")
- return "", "", "", [], [], "", ""
+ return "", "", "", "", [], [], "", ""
 
 def nameChange():
  # askdirectory()檔案對話框, 選擇目錄, 返回目錄名
  path = filedialog.askdirectory()
  if path == "":
  messagebox.showinfo(title="錯誤", message="請選擇路徑!" + "\n")
  else:
- cbtn.config(state=tk.DISABLED)
+ cbtn_deltext.config(state=tk.DISABLED)
+ cbtn_dlcover.config(state=tk.DISABLED)
  btn.config(state=tk.DISABLED)
  btn['text'] = "等待完成"
  text.insert(tk.END, "選擇路徑: " + path + "\n")
@@ -210,10 +213,10 @@ def nameChange():
  else:
  #print('Processing: ' + code)
  text.insert(tk.END, 'Processing: ' + code + '\n')
- r_status, title, circle, cvList, authorList, work_age, release_date = match_code(code)
+ r_status, img_url, title, circle, cvList, authorList, work_age, release_date = match_code(code)
  # 如果順利爬取網頁訊息
  if r_status == 200 and title and circle:
- if var1.get():
+ if deltext.get():
  # 刪除title中的【.*?】
  title = re.sub(u"\\【.*?】", "", title)
 
@@ -244,6 +247,19 @@ def nameChange():
  new_name = new_name.replace("cv", cv[1:])
  else:
  new_name = new_name.replace("(CV. cv)", "")
+
+
+ # 要下載封面且是資料夾
+ if dlcover.get() and img_url and os.path.isdir(os.path.join(path, file)):
+ try: # 嘗試下載封面
+ store_path = os.path.join(path, file, "cover.jpg")
+ if not os.path.isfile(store_path):
+ text.insert(tk.END, " 下載封面...\n")
+ urllib.request.urlretrieve(img_url, store_path)
+ else:
+ text.insert(tk.END, "**封面已存在，跳過下載!\n")
+ except os.error as err:
+ text.insert(tk.END, "**下載封面過程中出現錯誤!\n")
 
  # 將Windows文件名中的非法字元替換
  # re.sub(pattern, repl, string)
@@ -278,7 +294,8 @@ def nameChange():
  text.insert(tk.END, "*******完成!*******\n\n\n\n")
  tk.messagebox.showinfo(title="提示", message="完成!")
 
- cbtn.config(state=tk.NORMAL)
+ cbtn_deltext.config(state=tk.NORMAL)
+ cbtn_dlcover.config(state=tk.NORMAL)
  btn.config(state=tk.NORMAL)
  btn['text'] = "選擇路徑"
 
@@ -296,9 +313,9 @@ def thread_it(func, *args):
 
 
 root = tk.Tk() # 實例化object，建立視窗root
-root.title('DLsite重命名工具 v2.0') # 給視窗的標題取名字
+root.title('DLsite重命名工具 v3.0') # 給視窗的標題取名字
 root.eval('tk::PlaceWindow . center')
-root.geometry('300x375') # 設定視窗的大小(橫向 * 縱向)
+root.geometry('350x450') # 設定視窗的大小(橫向 * 縱向)
 
 text = tk.Text(root)
 text.pack()
@@ -359,13 +376,17 @@ def thread_it(func, *args):
  text.insert(tk.END, "**使用默認命名模板:\n")
  text.insert(tk.END, " workno title \n")
 
-var1 = tk.IntVar() # 定義var1整型變數用來存放選擇行為返回值
-cbtn = tk.Checkbutton(root, text='去除title中【】之間的內容', variable=var1,
+deltext = tk.IntVar() # 定義整數變數用來存放選擇行為返回值
+dlcover = tk.IntVar()
+cbtn_deltext = tk.Checkbutton(root, text='去除title中【】之間的內容', variable=deltext,
  onvalue=1, offvalue=0) # 傳值原理類似於radiobutton物件
+cbtn_dlcover = tk.Checkbutton(root, text='下載封面', variable=dlcover,
+ onvalue=1, offvalue=0) 
 
 btn = tk.Button(root, text='選擇路徑', command=lambda: thread_it(nameChange))
 
 btn.pack()
-cbtn.pack()
+cbtn_deltext.pack()
+cbtn_dlcover.pack()
 
 root.mainloop()