日に日に分からんことが増えていく…

プログラマー初心者としての日々を学んでいることを記録していく。

JSON形式のAPIの情報をExcelに投入

#!/usr/bin/env python
# coding:utf-8

import openpyxl as px
import requests
import json
import re

URL = "[API URL名]"
fname = "ファイル元"
num_string_pattern = re.compile(r'\d+')
itemid_string_pattern = re.compile(r'\w{3}\d{4}')
itemid_string_pattern2 = re.compile(r'\w{1}\d{4}')


def check_items(keyword, url):
    params = {'q': keyword}
    r = requests.get(URL, params=params)
    dic = json.loads(r.text)
    total = dic.get('result')get('total')
    return total > 0


def digits_check(keyword):
    if re.match(num_string_pattern, keyword) or re.match(itemid_string_pattern, keyword) or re.match(itemid_string_pattern2, keyword):
        return True
    else:
        return False


def run(fname):
    try:
        wb = px.load_workbook(fname)
        print(wb.get_sheet_names())
        ws = wb.get_sheet_by_name('データセット1')
        _write_sheet(ws)
    except:
        raise
    else:
        wb.save(fname)


def _write_sheet(sheet):
    for row in range(2, 10002):
        keyword = sheet['A{}'.format(row)].value
        keyword = str(keyword)
        print(keyword)
        sheet['I{}'.format(row)].value = digits_check(keyword).numerator
        sheet['J{}'.format(row)].value = digits_check(keyword).numerator


def main():
    run(fname)


if __name__ == '__main__':
    main()

www.lisz-works.com