import pandas as pd
import matplotlib.pyplot as plt
from PIL import Image
from bs4 import BeautifulSoup as bs
from datetime import datetime, timedelta
import requests, pymysql, ftputil, hashlib, os, sys

 
def file_type(x): # 그누보드의 bf_type 값을 반환하는 함수입니다.
    return {'gif' : '1', 'jpeg' : '2', 'jpg' : '2', 'png' : '3', 'swf' : '4', 'psd' : '5',
            'bmp' : '6', 'tif' : '7', 'tiff' : '7', 'jpc' : '9', 'jp2' : '10', 'jpx' : '11',
            'jb2' : '12', 'swc' : '13', 'iff' : '14', 'wbmp' : '15', 'xbm' : '16'}.get(x.lower(), '0')

 
def file_upload(filename, bf_file): # FTP를 이용하여 파일을 업로드하는 함수입니다.
    with ftputil.FTPHost('FTP주소', 'FTP아이디', 'FTP비번') as fh:
        fh.chdir('/web/data/file/board')
        fh.upload(filename, bf_file, callback = None)
    return

 
def get_filename(filename): # 파일명을 변환하는 함수입니다.
    ms = datetime.now().microsecond
    encoded_name = filename.encode('utf-8')
    result = f'{ms}_{hashlib.sha1(encoded_name).hexdigest()}'
    return result

 
def board_write(board, subject, content, mb_id, nickname, file_list = None):
    # MySQL connection 및 cursor 생성
    conn = pymysql.connect(host = 'DB의URL을입력하세요',
                           user = 'DB의아이디를입력하세요',
                           password = 'DB의패스워드를입력하세요',
                           db = 'DB명을입력하세요',
                           charset = 'utf8')
    curs = conn.cursor()
 
    # 작성글 INSERT
    sql = f"select wr_num from g5_write_{board}"
   
    curs.execute(sql)
    wr_num = str(int(curs.fetchone()[0]) - 1)
    print(wr_num)
    now = datetime.today().strftime('%Y-%m-%d %H:%M:%S') # 그누보드의 날짜 형식 준수 (ex: 2021-04-05 23:45:15)
    sql = f"insert into g5_write_{board} set wr_num = {wr_num}, \
          wr_reply = '', wr_comment = 0, ca_name = '', wr_option = 'html1', wr_subject = '{subject}', \
          wr_content = '{content}', wr_link1 = '', wr_link2 = '', \
          wr_link1_hit = 0, wr_link2_hit = 0, wr_hit = 1, wr_good = 0, wr_nogood = 0, \
          mb_id = '{mb_id}', wr_password = '', wr_name = '{nickname}', wr_email = '', wr_homepage = '', \
          wr_datetime = '{now}', wr_last = '{now}', wr_ip = '111.111.111.111', \
          wr_1 = '', wr_2 = '', wr_3 = '', wr_4 = '', wr_5 = '', \
          wr_6 = '', wr_7 = '', wr_8 = '', wr_9 = '', wr_10 = '', \
          wr_comment_reply = '', wr_facebook_user = '', wr_twitter_user = '', \
          as_re_name = '', as_tag = '', as_map = '', as_icon = '', as_thumb = '', as_video = ''"
    curs.execute(sql)
 
    # 부모 아이디에 UPDATE
    sql = f"select wr_id from g5_write_{board}"
    curs.execute(sql)
    wr_id = str(curs.fetchall()[-1][0])
    print(f"wr_id : {wr_id}")
    sql = f"update g5_write_{board} set wr_parent = {wr_id} where wr_id = {wr_id}"
    curs.execute(sql)
 
    # 새글 INSERT
    sql = f"insert into g5_board_new ( bo_table, wr_id, wr_parent, bn_datetime, mb_id ) values \
          ( '{board}', '{wr_id}', '{wr_id}', '{now}', '{mb_id}' )"
    curs.execute(sql)
 
    # 게시글 1 증가
    sql = f"select bo_count_write from g5_board where bo_table = '{board}'"
    curs.execute(sql)
    bo_count_write = str(int(curs.fetchone()[0]))
    print(curs.fetchall())
    print(bo_count_write)
    sql = f"update g5_board set bo_count_write = {bo_count_write} + 1 where bo_table = '{board}'"
    curs.execute(sql)
 
    # 파일 업로드 및 관련 정보를 테이블에 저장
    if not file_list or file_list == [''] or file_list == []: # 첨부파일이 없는 경우에는 스크립트를 중단합니다.
        conn.close()
        sys.exit()
 
    file_count = len(file_list)
    for cnt, file in enumerate(file_list):
        ext = os.path.splitext(file)[1].lstrip('.')
        bf_file = f'{get_filename(file)}.{ext}'
        file_upload(file, bf_file)
        type = file_type(ext)
        if type != '0': # 이미지 파일의 경우 가로 및 세로를 구하고, 그 외의 경우에는 0을 대입합니다.
            im = Image.open(file)
            width, height = im.size
        else:
            width, height = 0, 0
        size = os.path.getsize(file)
        sql = f"insert into g5_board_file set bo_table = '{board}', wr_id = '{wr_id}', \
              bf_no = '{cnt}', bf_source = '{file}', bf_file = '{bf_file}', \
              bf_content = '', bf_download = 0, bf_filesize = '{size}', \
              bf_width = '{width}', bf_height = '{height}', bf_type = '{type}', bf_datetime = '{now}'"
        curs.execute(sql)
       
    # 파일의 개수를 게시물에 업데이트
    sql = f"update g5_write_board set wr_file = '{file_count}' where wr_id = '{wr_id}'"
    curs.execute(sql)
 
    # MySQL connection 닫기
    conn.close()
    return

 
def alp2num(string): # 스트링에서 숫자를 제외한 문자를 삭제하는 함수
    return int(''.join([x for x in string if x.isdecimal()]))

 
def parse_naver_weather(elements): # 네이버날씨를 파싱하여 오전/오후의 기온 및 강수확률을 반환하는 함수
    data = []
    for e in elements:
        inners = e.findAll('span', {'class':'weather_inner'})
        rainAM = inners[0].find('span', {'class':'rainfall'}).text
        rainPM = inners[1].find('span', {'class':'rainfall'}).text
        tempAM = e.find('span', {'class':'lowest'}).find(text=True, recursive=False)
        tempPM = e.find('span', {'class':'highest'}).find(text=True, recursive=False)
        foo = [rainAM, tempAM, rainPM, tempPM]
        foo = list(map(alp2num, foo)) # 각 값에서 숫자를 제외한 문자열을 제거합니다.
        data.append(foo[:2]) # 오전과 오후의 데이터를 리스트에서 별개의 요소로 처리합니다.
        data.append(foo[2:])
    return data

 
def main():
    # 네이버 날씨를 크롤링합니다.
    html = requests.get('https://search.naver.com/search.naver?where=nexearch&sm=top_hty&fbm=0&ie=utf8&query=%EC%88%98%EC%9B%90+%EB%82%A0%EC%94%A8')
    base_url = 'https://m.search.naver.com/p/csearch/content/nqapirender.nhn?where=nexearch&pkid=227&u1=02111129&key='
    sources = {'weather', 'accuWeather', 'twcWeather', 'weatherNews'} # URL 마지막에 붙는 파라미터입니다.
    urls = {base_url + x for x in sources}
    for cnt, url in enumerate(urls, 1):
        r = requests.get(url)
        html = str(r.json()['weekly'])
        soup = bs(html, 'html5lib')
        lis = soup.findAll('li')
        df = pd.DataFrame(parse_naver_weather(lis))
        globals()['df{}'.format(cnt)] = df # 동적으로 변수를 선언하여 결과값을 대입합니다.
 
    # 각 DataFrame의 평균을 구하는 등 적절히 조작합니다.
    result = (df1 + df2 + df3 + df4) / 4
    result = result.dropna().round().astype(int)
    result.columns = ['Rain', 'Temp']
 
    # 그래프 X축으로 사용할 날짜를 구합니다(오늘 포함 총 8일).
    dt = datetime.now().date()
    dates8 = [(dt + timedelta(days=i)).strftime('%m/%d') for i in range(8)]
    x_axis = [x for pair in zip(dates8, dates8) for x in pair] # 오전/오후이기 때문에 두 번씩 넣어줍니다.
 
    # Matplotlib를 이용하여 그래프로 표현합니다.
    plt.style.use('ggplot')
    fig = plt.figure()
    ax = fig.add_subplot()
    ax.xaxis.set_ticks(range(16))
    ax.set_xticklabels(x_axis)
    for i, tick in enumerate(ax.xaxis.get_ticklabels()): # X축의 날짜를 한 칸씩 건너서 표시합니다.
        if i % 2 != 0:
            tick.set_visible(False)
    plt.plot(result['Temp'], marker='o', markersize=10, color='red', label='Temperature')
    bbox = dict(boxstyle='round, pad=0.2', fc='skyblue', alpha=0.5)
    for x, txt in enumerate(result['Rain']):
        color = 'blue' if result['Rain'][x] >= 50 else 'white' # 강수확률 50% 이상이면 파란색 글씨로 표기합니다.
        ax.annotate(f'{txt}%', xy=(x, result['Temp'][x]), xytext=(5, -8),
                    color=color, textcoords='offset points', bbox=bbox,
                    horizontalalignment='left', verticalalignment='top')
    plt.legend(loc='best')
    plt.xticks(rotation=75)
    plt.savefig('forecast.png')
 
    # 그누보드에 게시글을 작성합니다.
    today = dt.strftime('%Y-%m-%d')
    board = 'board'
    subject = f'{today}자 기상관측 4개 기관 일기예보 분석'
    content = '기상청/아큐웨더/웨더뉴스/웨더채널의 데이터를 수집하여 이를 토대로 작성한 예보입니다.'
    mb_id = '그누보드아이디를입력하세요'
    nickname = '그누보드닉네임을입력하세요'
    file_list = ['forecast.png']
    board_write(board, subject, content, mb_id, nickname, file_list)

 
if __name__ == "__main__":
    main()