#!/usr/bin/python

# -*- coding:utf-8 -*-

# @filename: search

# @author：wwx399777 wuweiji

# @date: 2018/1/25 9:35

import pymysql

import pyecharts

from datetime import datetime

import pandas as pd

import numpy as np

import matplotlib as mpl

import re

def connect_mysql(sql, oper_type="select", data_l=None):

    conn = pymysql.connect(host='localhost', user="root", password="***",

                           database="work", port=3306)

    cur = conn.cursor()

    if oper_type == "insert":

        cur.executemany(sql, data_l)

        conn.commit()

    else:

        cur.execute(sql)

    result = cur.fetchall()

    conn.close()

    return result

def data_analysis():

    sql = """

    SELECT p.name, p.email, p.memo from personal_userinfo p

    """

    x = connect_mysql(sql)

    print(type(x), x)

    x = list(x[2])

    print(x)

    return x

data_analysis()

def create_list(file_name):

    ls = []

    if not re.findall(file_name, r".txt"):

        file_name += ".txt"

    with open(file_name) as f:

        for i in f:

            name, email, memo = i.strip().split(",")

            ls.append([name, email, memo, datetime.utcnow()])

    return ls

def data_insert():

    sql = """

    INSERT INTO WORK.personal_userinfo (name, email, memo, date_time) VALUES (%s, %s, %s, %s)

    """

    data_l = create_list("person_info")

    connect_mysql(sql, oper_type="insert", data_l=data_l)

# data_insert()