-
Notifications
You must be signed in to change notification settings - Fork 40
/
Copy pathpublish_all.py
181 lines (147 loc) · 6.64 KB
/
publish_all.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
import os.path
import traceback
import selenium
from selenium import webdriver
from publisher.alicloud_publisher import alicloud_publisher
from publisher.cnblogs_publisher import cnblogs_publisher
from publisher.csdn_publisher import csdn_publisher
from publisher.cto51_publisher import cto51_publisher
from publisher.infoq_publisher import infoq_publisher
from publisher.oschina_publisher import oschina_publisher
from publisher.segmentfault_publisher import segmentfault_publisher
from publisher.toutiao_publisher import toutiao_publisher
from publisher.txcloud_publisher import txcloud_publisher
from publisher.zhihu_publisher import zhihu_publisher
from publisher.jianshu_publisher import jianshu_publisher
from publisher.juejin_publisher import juejin_publisher
from publisher.mpweixin_publisher import mpweixin_publisher
from utils.file_utils import list_all_files, list_files, write_to_file, read_head
from utils.yaml_file_utils import read_common
last_published_file_name = 'last_published.txt'
common_config = read_common()
driver_type = common_config['driver_type']
content_dir = common_config['content_dir']
if driver_type == 'chrome':
# 启动浏览器驱动服务
service = selenium.webdriver.chrome.service.Service(common_config['service_location'])
# Chrome 的调试地址
debugger_address = common_config['debugger_address']
# 创建Chrome选项，重用现有的浏览器实例
options = selenium.webdriver.chrome.options.Options()
options.page_load_strategy = 'normal' # 设置页面加载策略为'normal' 默认值, 等待所有资源下载,
options.add_experimental_option('debuggerAddress', debugger_address)
# 使用服务和选项初始化WebDriver
driver = webdriver.Chrome(service=service, options=options)
elif driver_type == 'firefox':
# 启动浏览器驱动服务
service = selenium.webdriver.firefox.service.Service(common_config['service_location'],
service_args=['--marionette-port', '2828',
'--connect-existing'])
# 创建firefox选项，重用现有的浏览器实例
options = selenium.webdriver.firefox.options.Options()
options.page_load_strategy = 'normal' # 设置页面加载策略为'normal' 默认值, 等待所有资源下载,
driver = webdriver.Firefox(service=service, options=options)
driver.implicitly_wait(10) # 设置隐式等待时间为15秒
all_sites = ['csdn',
'jianshu',
'juejin',
'segmentfault',
'oschina',
'cnblogs',
'zhihu',
'cto51',
'infoq',
'txcloud',
'alicloud',
'toutiao',
'mpweixin']
def publish_to_platform(platform, driver, content=None):
"""
发布到指定平台的封装函数
"""
try:
globals()[platform + '_publisher'](driver, content) # 动态调用对应平台的发布函数
except Exception as e:
print(platform, "got error")
traceback.print_exc() # 打印完整的异常跟踪信息
print(e)
if content:
save_last_published_file_name(os.path.basename(content))
def publish_to_all_platforms(driver, content=None):
"""
发布到所有平台的封装函数
"""
for platform in all_sites:
if platform in common_config['enable'] and common_config['enable'][platform]:
publish_to_platform(platform, driver, content)
# 在需要的时候关闭浏览器，不要关闭浏览器进程
driver.quit()
def save_last_published_file_name(filename):
write_to_file(filename, last_published_file_name)
if __name__ == '__main__':
while True:
# 选择要发布的内容
print("选择你要发布的博客,输入序号:")
# file_list = list_all_files(content_dir, ".md")
file_list = list_files(content_dir, ".md")
for index, file_name in enumerate(file_list):
print(str(index) + ":" + os.path.basename(file_name))
print("上次发布的博客是: " + read_head(last_published_file_name))
file_choice = input("\n请选择: ")
print("")
if len(file_list) > int(file_choice) >= 0:
file_path = file_list[int(file_choice)]
print("你要发布的文章是:", file_path)
while True:
print("选择你要发布的平台:\n")
print("1. 全部")
print("2. csdn")
print("3. jianshu")
print("4. juejin")
print("5. segmentfault")
print("6. oschina")
print("7. cnblogs")
print("8. zhihu")
print("9. cto51")
print("10. infoq")
print("11. txcloud")
print("12. alicloud")
print("13. toutiao")
print("14. mpweixin")
print("0. 退出")
choice = input("\n请选择: ")
print("")
if choice == '1':
publish_to_all_platforms(driver, file_path)
elif choice == '2':
publish_to_platform('csdn', driver, file_path)
elif choice == '3':
publish_to_platform('jianshu', driver, file_path)
elif choice == '4':
publish_to_platform('juejin', driver, file_path)
elif choice == '5':
publish_to_platform('segmentfault', driver, file_path)
elif choice == '6':
publish_to_platform('oschina', driver, file_path)
elif choice == '7':
publish_to_platform('cnblogs', driver, file_path)
elif choice == '8':
publish_to_platform('zhihu', driver, file_path)
elif choice == '9':
publish_to_platform('cto51', driver, file_path)
elif choice == '10':
publish_to_platform('infoq', driver, file_path)
elif choice == '11':
publish_to_platform('txcloud', driver, file_path)
elif choice == '12':
publish_to_platform('alicloud', driver, file_path)
elif choice == '13':
publish_to_platform('toutiao', driver, file_path)
elif choice == '14':
publish_to_platform('mpweixin', driver, file_path)
elif choice == '0':
break
else:
print("无效的输入，请重新输入。")
else:
print("无效的选择")