wik.py

import json
import requests
from bs4 import BeautifulSoup as Soup
from googleapiclient.discovery import build

import config


def correct(text):
    text = text.replace('\n', ' ')
    response = requests.get('http://csearch.naver.com/dcontent/spellchecker.nhn', params={
        '_callback': '',
        'q': text,
    }, headers={
        'User-Agent': config.user_agent,
    })
    result = json.loads(response.text[1:-2])
    soup = Soup(result['message']['result']['html'], 'lxml')
    return soup.get_text()


def translate_ko_en(text):
    service = build('translate', 'v2', developerKey=config.google_dev_key)
    result = service.translations().list(
        source='ko',
        target='en',
        q=[text],
    ).execute()

    return result['translations'][0]['translatedText']


def wik(line, cell):
    corrected = correct(cell)
    print(corrected)
    translated = translate_ko_en(corrected)
    print(translated)


def load_ipython_extension(shell):
    shell.register_magic_function(wik, 'cell')