# coding: utf-8
import os
import sys
def format_file(filename, str1, str2):
"""
文件内容的替换功能
:return:
"""
with open(filename, 'r') as f:
var_object = f.read()
if "gitalk" not in var_object:
var_object = var_object.replace(str1, str2)
# print(var_object)
f = open(filename, "w")
f.write(var_object)
if __name__ == "__main__":
if len(sys.argv) == 3:
version, u_type = sys.argv[1], sys.argv[2]
else:
print("Usage: 参数个数为%s - 错误，应该改为3" % len(sys.argv))
sys.exit(-1)
tag = True
if u_type == "index":
tag = False
# if version == "home":
# filename = "_book/index.html"
# else:
# filename = "_book/docs/%s/index.html" % version
# str1 = """
#
#
# """
# str2 = """
#
#
#
# """
elif u_type == "book":
if version == "home":
filename = "book.json"
tag = False
else:
filename = "docs/%s/book.json" % version
str1 = "https://github.com/apachecn/AiLearning/blob/master"
str2 = "https://github.com/apachecn/AiLearning/blob/master/docs/%s" % version
elif u_type == "powered":
if version == "home":
filename = "node_modules/gitbook-plugin-tbfed-pagefooter/index.js"
else:
filename = "docs/%s/node_modules/gitbook-plugin-tbfed-pagefooter/index.js" % version
str1 = "powered by Gitbook"
str2 = "由 ApacheCN 团队提供支持"
elif u_type == "gitalk":
if version == "home":
filename = "node_modules/gitbook-plugin-tbfed-pagefooter/index.js"
else:
filename = "docs/%s/node_modules/gitbook-plugin-tbfed-pagefooter/index.js" % version
str1 = """ var str = ' \\n\\n'"""
str2 = """
var str = '\\n\\n'+
'\\n
'+
'\\n'+
'\\n '+
'\\n '+
'\\n '+
'\\n '+
'\\n '+
'\\n'+
'\\n '+
'\\n'+
'\\n '+
'\\n '+
'\\n
'+
'\\n'+
'\\n'+
'\\n'+
'\\n
'
str += '\\n\\n'+
'\\n'+
'\\n
'+
'\\n '+
'\\n '+
'\\n
'+
'\\n '+
'\\n
