import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import odeint
from mpl_toolkits.mplot3d.axes3d import Axes3D
import matplotlib.animation as animation


# === define the lorenz system ===
# x, y, and z make up the system state
# t is time,
# sigma, rho, beta are the system parameters
def lorenz_system(current_state, t):
    x, y, z = current_state

    dx_dt = sigma * (y - x)
    dy_dt = x * (rho - z) - y
    dz_dt = x * y - beta * z

    return [dx_dt, dy_dt, dz_dt]


# define the initial system state, system parameters
initial_state = [-0.1, 0.01, 0.08]
sigma = 10.
rho = 50.
beta = 3.

start_time = 0
end_time = 10
num_points = 100 * (end_time - start_time)
time_points = np.linspace(start_time, end_time, num_points)

# === generate datas =========
xyz = odeint(lorenz_system, initial_state, time_points)
data = np.array(xyz).T * 0.01  # scale down to 0.01
print(data)

print(data[0])

import leancloud

leancloud.init("xxx", "yyy")

TestObject = leancloud.Object.extend('my_log')
test_object = TestObject()
test_object.set('col1', 'a')
test_object.set('col2', 'b')
test_object.set('col3', 'c')
test_object.save()

# pip install icalendar

# %%
import pandas as pd

df = pd.read_csv('1.csv',  
                encoding='gbk',
                header=0, 
                parse_dates=[0,1], 
                date_parser=lambda x: pd.to_datetime(x, format='%Y/%m/%d %H:%M'))

print(df)


# %%
from icalendar import Calendar, Event

cal = Calendar()
cal.add('version', '2.0')


for row in df.itertuples(index=False):
    stime, etime, thing = row[0], row[1], row[2]

    # print(row[0], row[1], row[2])

    event = Event()
    event.add('summary', thing)
    event.add('dtstart', stime)
    event.add('dtend', etime)
    # print(event)

    cal.add_component(event)

    txt = cal.to_ical()
    print(str(txt, encoding='utf8'))


# %%
with open('1.ics', 'wb') as f:
    f.write(txt)

# xyz_a.py
# %%
import os
import shutil

from hashlib import md5


# 创建 md5 对象
def encrypt_md5(s):
    new_md5 = md5()
    new_md5.update(s.encode(encoding='utf-8'))
    return new_md5.hexdigest()


# old_root = os.getcwd()    # 待处理文件的根目录
old_root = "E:\\"
new_root = "D:\\0\\"
exts = ['.xyz']


for parent, dirnames, filenames in os.walk(old_root):
    for filename in filenames:
        extension = os.path.splitext(filename)[1]
        
        if extension.lower() in ext:

            old_dir = os.path.join(parent, filename)
            new_name = encrypt_md5(old_dir) + '_' + filename   # 修改文件名
            new_dir = os.path.join(new_root, new_name)
            
            print(f'{old_dir};; {new_name};; {new_dir}')
            
            # shutil.copyfile(old_dir, new_dir)    # 复制到新文件夹

# %%
import os
import fitz
from pprint import pprint

fapiao_dir = 'src/'

zhantiedan = 'zhantiedan_blank.pdf'
ztd_pdf = fitz.open(zhantiedan)

new_name = 'fapiao_dayin.pdf'
new_pdf = fitz.open()


def fapiao_zhantie(fapiao=''):
    
    print(f'\n=== fapiao pdf file:\n    {fapiao}')
    src_pdf = fitz.open(fapiao)

    for p, page in enumerate(src_pdf):
        pixmap = page.get_pixmap(dpi=300)
        pixmap.save(f'{fapiao}_{str(p)}.png')
        
        new_pdf.insert_pdf(ztd_pdf)
        end_page = new_pdf[-1]
        end_page.insert_image((100, 100, 700, 500), pixmap=pixmap)

    src_pdf.close()

    new_pdf.save(new_name)
    print(f'    {fapiao} :\n*** Done! ***\n')


for dir, folders, files in os.walk(fapiao_dir):
    for file in files:
        path = os.path.join(dir, file)
        print(path)
        if path[-4:] == '.pdf':
            fapiao_zhantie(path)

ztd_pdf.close()
new_pdf.close()

import fitz
import re
from pprint import pprint

pdf_name = 'xxx-en.pdf'
print(f'Source pdf file: {pdf_name} \n')
src_pdf = fitz.open(pdf_name)
new_pdf = fitz.open()

for p, page in enumerate(src_pdf):
    print(f'\n- translating PAGE -{p}- ...')

    # 1.1 创建大小相同的新页面
    new_page = new_pdf.new_page(width=page.rect.width, height=page.rect.height)

    blocks = page.get_text('dict')['blocks']

    # 1.2 图片
    img_blks = [b for b in blocks if b['type'] == 1]
    for img in img_blks:
        # pprint(img)
        new_page.insert_image(img['bbox'], stream=img['image'])

    # 1.3 文字
    txt_blks = [b for b in blocks if b['type'] != 1]
    for txt in txt_blks:
        text_tmp = ''.join([s['text'] for l in txt['lines'] for s in l['spans']])
        text_tmp = re.sub('[@#$%^&*\'\"\n\r\t]', ' ', text_tmp).strip()

        if text_tmp:
            # print(txt['bbox'], text_tmp)

            text_translate = '中国 ' + text_tmp
            # text_translate = youdao(text_tmp)
            new_page.insert_textbox(txt['bbox'], text_translate,
                                    fontsize=6,
                                    fontname='simhei',
                                    fontfile=r'C:\Windows\Fonts\simhei.ttf')

    # if p == 1:
    #     break

new_name = pdf_name.replace('.pdf', '-zh.pdf')
new_pdf.save(new_name)

print('\n------Done!-------')