content.py 4.19 KB
import csv
import json
import time
import random
import os
import re
import pandas as pd
import numpy as np

from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from hanspell import spell_checker
from bs4 import BeautifulSoup
from datetime import datetime, timedelta

BASE_DIR = os.path.dirname(os.path.realpath(__file__))

def sleeptime():
    time.sleep(random.randint(1, 3))

def Click(xpath, driver):
    element = driver.find_element_by_xpath(xpath)
    driver.execute_script("arguments[0].click();", element)
    sleeptime()

def TextPreprocess(text):
    text = re.sub(pattern='[^\w\s]', repl='', string=text)
    text = re.sub(pattern='\n', repl='. ', string=text)

    spelled_sent = spell_checker.check(text)
    text = spelled_sent.checked

    return text

def GetData():
    login_info = {
        'userID' : '**********',
        'userpw' : '**********'
    }

    options = webdriver.ChromeOptions()
    options.add_argument('headless')
    options.add_argument('no-sandbox')
    options.add_argument('window-size=1920x1080')
    options.add_argument('disable-gpu')
    options.add_argument('disable-dev-shm-usage')
    options.add_argument('lang=ko_KR')
    options.add_argument('user-agent=Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/65.0.3325.183 Safari/537.36 Vivaldi/1.96.1147.47')

    # driver = webdriver.Chrome(r'C:\Users\E_N__\Desktop\chromedriver.exe', options=options)
    driver = webdriver.Chrome(BASE_DIR + '/chromedriver.exe', options=options)

    driver.get('about:blank')
    driver.execute_script("Object.defineProperty(navigator, 'plugins', {get: function() {return[1, 2, 3, 4, 5];},});")
    sleeptime()
    driver.get('https://everytime.kr/login')

    sleeptime()
    driver.find_element_by_name('userid').send_keys(login_info['userID'])
    driver.find_element_by_name('password').send_keys(login_info['userpw'])
    driver.find_element_by_class_name('submit').click()
    sleeptime()

    # 국제캠 자게
    sleeptime()
    yesterday = (datetime.today() - timedelta(1)).strftime('%m/%d %H:%M')
    print(yesterday)
    swt = True
    page = 0

    post_df = pd.DataFrame(columns=['title', 'content'])
    while swt:
        if page < 1:
            Click('//*[@id="submenu"]/div/div[2]/ul/li[1]/a', driver)
            page += 1
        else:
            if page == 1:
                Click('//*[@id="container"]/div[2]/div[2]/a', driver)
                page += 1
            elif page == 2:
                Click('//*[@id="container"]/div[2]/div[2]/a[2]', driver)
                page += 1
            else:
                Click('//*[@id="container"]/div[2]/div[2]/a[3]', driver)

        html = driver.page_source
        soup = BeautifulSoup(html, 'html.parser')

        TitleList = soup.select('#container > div.wrap.articles > article > a > h2')
        DateList = soup.select('#container > div.wrap.articles > article > a > time')

        for post in zip(TitleList, DateList):
            title = TextPreprocess(post[0].text)

            try:
                Click("//h2[contains(text(), '{}')]".format(title), driver)
            except NoSuchElementException:
                continue
            content = driver.find_element_by_xpath('//*[@id="container"]/div[2]/article/a/p').text
            driver.back()
            sleeptime()

            if not (post_df['title'] == title).any():
                content = TextPreprocess(content)
                post_df = post_df.append(pd.DataFrame([[title, content]],
                                                      columns=['title', 'content']))
            if post[1].text <= yesterday:
                break
        break

    post_df.to_csv('data.csv', mode='w', encoding='utf-8-sig', index=False)
    print("CVS file saved")

    with open('data.json', 'w+', encoding='utf-8-sig') as json_file:
        for post in zip(post_df['title'].tolist(), post_df['content'].tolist()):
            json.dump({
                "document" :
                    {
                        "type" : "PLAIN_TEXT",
                        "content" : post[0] + post[1]
                    },
                "encodingType" : "UTF8"
            }, json_file, ensure_ascii=False)
    print("JSON file saved")

GetData()