# Copyright 2022 ljmold.cn All rights reserved
#
# Authors: Carry Jin 
# 将电子书中的繁体中文转换为简体中文
import opencc
import os
import zipfile
import shutil

def test_convert():
    """
     s2t.json Simplified Chinese to Traditional Chinese 簡體到繁體
    t2s.json Traditional Chinese to Simplified Chinese 繁體到簡體
    """
    converter = opencc.OpenCC('s2t.json')
    print(converter.convert('汉字acb'))  # 漢字

    converter = opencc.OpenCC('t2s.json')
    print(converter.convert('漢字123'))  # 漢字

def unzip_dir(zipfilename, unzipdirname):
    """解压zip文件"""
    fullzipfilename = os.path.abspath(zipfilename)
    fullunzipdirname = os.path.abspath(unzipdirname)
    print("Start to unzip file %s to folder %s ..." % (zipfilename, unzipdirname))
    # Check input ...
    if not os.path.exists(fullzipfilename):
        print("Dir/File %s is not exist, Press any key to quit..." % fullzipfilename)
        inputStr = input()
        return
    if not os.path.exists(fullunzipdirname):
        os.mkdir(fullunzipdirname)
    else:
        if os.path.isfile(fullunzipdirname):
            print("File %s is exist, are you sure to delet it first ? [Y/N]" % fullunzipdirname)
            while 1:
                inputStr = input()
                if inputStr == "N" or inputStr == "n":
                    return
                else:
                    if inputStr == "Y" or inputStr == "y":
                        os.remove(fullunzipdirname)
                        print("Continue to unzip files ...")
                        break

                        # Start extract files ...
    srcZip = zipfile.ZipFile(fullzipfilename, "r")
    for eachfile in srcZip.namelist():
        if eachfile.endswith('/'):
            # is a directory
            print('Unzip directory %s ...' % eachfilename)
            os.makedirs(os.path.normpath(os.path.join(fullunzipdirname, eachfile)))
            continue
        print("Unzip file %s ..." % eachfile)
        eachfilename = os.path.normpath(os.path.join(fullunzipdirname, eachfile))
        eachdirname = os.path.dirname(eachfilename)
        if not os.path.exists(eachdirname):
            os.makedirs(eachdirname)
        fd = open(eachfilename, "wb")
        fd.write(srcZip.read(eachfile))
        fd.close()
    srcZip.close()
    print("Unzip file succeed!")

def zip_dir(dirname,zipfilename):
    filelist = []
    if os.path.isfile(dirname):
        filelist.append(dirname)
    else :
        for root, dirs, files in os.walk(dirname):
            for dir in dirs:
                filelist.append(os.path.join(root,dir))
            for name in files:
                filelist.append(os.path.join(root, name))

    zf = zipfile.ZipFile(zipfilename, "w", zipfile.zlib.DEFLATED)
    for tar in filelist:
        arcname = tar[len(dirname):]
        #print arcname
        zf.write(tar,arcname)
    zf.close()

def convert_file_to_chinese(file_path):
    """按行读取文件，存储到list集合中，转换元素的语言(繁体->简体)，将结果写回到文件中"""
    file_lines = []
    converter = opencc.OpenCC('t2s')

    with open(file_path, mode='r', encoding='utf-8') as f:
        for line in f.readlines():
            file_lines.append(converter.convert(line))

    with open(file_path, mode='w', encoding='utf-8') as f:
        f.writelines(file_lines)

def convert_epub_simplified(file_path):
    """将epub文件转繁体换成简体"""
    if not (os.path.exists(file_path) or os.path.isfile(file_path)):
        raise Exception("请检查文件路径：{}", file_path)

    dir_name, file_name = os.path.split(file_path)
    unzip_dir_path = dir_name + "/unzip"
    unzip_dir(epub_file_path, unzip_dir_path)

    files = find_content_files(unzip_dir_path)
    for file in files:
        convert_file_to_chinese(file)
    new_file_name = file_name[0:file_name.rindex(".")] + "-简体.epub"
    new_epub_file_path = os.path.join(dir_name, new_file_name)

    zip_dir(unzip_dir_path, new_epub_file_path)
    # os.remove(unzip_dir_path)
    shutil.rmtree(unzip_dir_path)

def find_content_files(folder_path):
    """查询文件夹中所有需要修改的文件的路径，返回路径的列表
    只转换epub文件的内容，搜索后缀为 'xhtml' 的文件
    """
    result_files = []
    for root, dirs, files in os.walk(folder_path):
        # for dir in dirs:
        #     filelist.append(os.path.join(root, dir))
        for name in files:
            if name.endswith('xhtml'):
                result_files.append(os.path.join(root, name))

    return result_files

if __name__ == '__main__':
    # 测试繁体简体转换
    # test_convert()
    epub_file_path = "/Users/jinzhao/Downloads/天字第一当.epub"
    convert_epub_simplified(epub_file_path)

drone-runner:
   image: drone/drone-runner-docker:1.8
   container_name: drone-runner
   ports:
     - "3000:3000"
   depends_on:
     - drone
   volumes:
     - /etc/localtime:/etc/localtime:ro
     - /var/run/docker.sock:/var/run/docker.sock:rw
   environment:
     - TZ=Asia/Shanghai
    ....
     - DRONE_CPU_PERIOD=100000
     - DRONE_CPU_QUOTA=150000
     - DRONE_MEMORY_LIMIT=500*1024*1024*8

cd 你的pg目录，就是data文件夹
mkdir -p ./9.4
mv ./data ./9.4/
# 最好先zip备份下数据
zip -r 9.4.zip 9.4
mkdir -p ./9.6/data
mkdir -p ./14/data
# 目录准备完毕，开始升级
#先升级到9.6
docker run --rm \
    -v $PWD:/var/lib/postgresql \
    tianon/postgres-upgrade:9.4-to-9.6 \
    --link
# 再升级到14
docker run --rm \
    -v $PWD:/var/lib/postgresql \
    tianon/postgres-upgrade:9.6-to-14 \
    --link

$ mkdir -p postgres-upgrade-testing
$ cd postgres-upgrade-testing
$ OLD='9.4'
$ NEW='9.5'

$ docker pull "postgres:$OLD"
$ docker run -dit \
    --name postgres-upgrade-testing \
    -v "$PWD/$OLD/data":/var/lib/postgresql/data \
    "postgres:$OLD"
$ sleep 5
$ docker logs --tail 100 postgres-upgrade-testing

$ # let's get some testing data in there
$ docker exec -it \
    -u postgres \
    postgres-upgrade-testing \
    pgbench -i -s 10

$ docker stop postgres-upgrade-testing
$ docker rm postgres-upgrade-testing

$ docker run --rm \
    -v "$PWD":/var/lib/postgresql \
    "tianon/postgres-upgrade:$OLD-to-$NEW" \
    --link

$ docker pull "postgres:$NEW"
$ docker run -dit \
    --name postgres-upgrade-testing \
    -v "$PWD/$NEW/data":/var/lib/postgresql/data \
    "postgres:$NEW"
$ sleep 5
$ docker logs --tail 100 postgres-upgrade-testing

$ # can now safely remove "$OLD"
$ sudo rm -rf "$OLD"

systemctl enable wg-quick@wg0
systemctl start wg-quick@wg0