『退屈なことはPythonにやらせよう』その10(12章:Excel)

#モジュールの読み込み(pip installが事前に必要)
import openpyxl

#既にあるワークブックの読み込み
wb = openpyxl.load_wrokbook("ワークブックのパス")
#新規のワークブックの作成をする場合
wb = openpyxl.Workbook()

#Do something awesome

#保存
wb.save("書き込み先のファイル名.xlsx")

#cell
sheet["A1"].value #A1の値を取得する
sheet.cell(row=1, column=3).value #x, y 座標を用いてセルを指定する(A1が(1,1)となる)

ref_obj = openpyxl.chart.References(sheet, min_col = int, min_row=int, max_col = int, max_row = int)
series_obj = openpyxl.chart.Series(ref_obj, title = "title whatever you want")
chart_obj = openpyxl.chart.BarChart()
chart_obj.append(series_obj)

#位置
chart_obj.y = int
chart_obj.x = int

#図のサイズ
chart_obj.w = int
chart_obj.h = int

#put figure on worksheet
sheet.add_chart(chart_obj)
wb.save("ファイルのパス")

#! python3
# multiplicationTable.py  -  コマンドラインから数字Nを受け取り、N×Nの掛け算の方を生成する

#debug
import logging
logging.basicConfig(level=logging.DEBUG,
format = " %(asctime)s - %(levelname)s - %(message)s")
logging.debug("プログラム開始")
logging.disable(logging.CRITICAL)

#checking_project
import sys, openpyxl
from openpyxl.utils import get_column_letter

number = int(sys.argv[1])
logging.debug("Nは{}です".format(number))

#preparing workbook and sheet
wb = openpyxl.Workbook()
sheet = wb.active


for i in range(1,number+1):
    #putting numbers at first level    
    row_cell = sheet.cell(row = (i+1), column=1)
    col_cell = sheet.cell(row = 1 , column = (i+1))
    row_cell.value = i
    col_cell.value = i

for i in range(1,number+1):
    #putting calculated numbers
    for j in range(1,number+1):
        cal_cell = sheet.cell(row=i+1, column=j+1)
        r_cell = sheet.cell(row=i+1, column = 1).value
        c_cell = sheet.cell(row = 1, column = j+1).value
        cal_cell.value = r_cell * c_cell
        logging.debug("{}は{}です".format(cal_cell, cal_cell.value))

wb.save("check12_1.xlsx")

#! python3
# check12_2.py  -  空行を挿入する
# Usage
# python check12_2.py <空行を挿入する行> <空行の数> <ワークシートのファイル名>

#debug
import logging
logging.basicConfig(level=logging.DEBUG,
format = " %(asctime)s - %(levelname)s - %(message)s")
logging.debug("プログラム開始")
logging.disable(logging.CRITICAL)

import sys, openpyxl
from openpyxl.utils import get_column_letter

N = int(sys.argv[1])
M = int(sys.argv[2])
logging.debug(N)

x_file =  sys.argv[3]

wb = openpyxl.load_workbook(x_file)
sheet = wb.active

wb_new = openpyxl.Workbook()
sheet_new = wb_new.active

for i in range(0, N):
    j = 1
    for cell_obj in list(sheet.rows)[i]:
        logging.debug("count is {}".format(j))
        logging.debug("cell_objは{}".format(cell_obj.value))
        sheet_new[(get_column_letter(j)+str(i+1))] = (cell_obj.value)
        logging.debug(sheet_new[(get_column_letter(j)+str(i+1))])
        j += 1

for i in range(N+M, sheet.max_row + M):
    j = 1
    for cell_obj in list(sheet.rows)[i-M]:
        logging.debug("count is {}".format(j))
        logging.debug("cell_objは{}".format(cell_obj.value))
        sheet_new[(get_column_letter(j)+str(i+1))] = (cell_obj.value)
        logging.debug(sheet_new[(get_column_letter(j)+str(i+1))])
        j += 1

wb_new.save("inserted_{}".format(x_file))

#! python3
# prepare_check12_3.py  -  横長のExcelファイルを生成する(縦3*横12)
#debug
import logging
logging.basicConfig(level=logging.DEBUG,
format = " %(asctime)s - %(levelname)s - %(message)s")
logging.debug("プログラム開始")
#logging.disable(logging.CRITICAL)

import openpyxl

wb = openpyxl.Workbook()
sheet = wb.active


for i in range(1,13):
    #putting numbers at first level    
    cell_1 = sheet.cell(row = 1, column= i)
    cell_2 = sheet.cell(row = 2, column = i)
    cell_3 = sheet.cell(row = 3, column = i)

    cell_1.value = "neko"
    cell_2.value = "cat"
    cell_3.value = "dog"

wb.save("ch12_3_pre.xlsx")

#! python3
# prepare_check12_3.py  -  横長のExcelファイルを生成する(縦3*横12)
#debug
import logging
logging.basicConfig(level=logging.DEBUG,
format = " %(asctime)s - %(levelname)s - %(message)s")
logging.debug("プログラム開始")
#logging.disable(logging.CRITICAL)

#モジュール読み込み
import openpyxl

#シートの読み込み
wb = openpyxl.load_workbook("ch12_3_pre.xlsx")
sheet = wb.active
#出力先シートの準備
wb_new = openpyxl.Workbook()
sheet_new = wb_new.active

#行と列の入れ替え
for i in range(1, sheet.max_row + 1):         #iが横
    for j in range(1, sheet.max_column + 1):  #jが縦

        #putting values to new sheet
        sheet_new.cell(row=j, column=i).value = sheet.cell(row = i, column = j).value  

#Save to new file
wb_new.save("t_ch12_3_pre.xlsx")

#! python3
# pre_check12_4.py  -  複数のテキストファイルを作成する


import os

os.makedirs("NEKO", exist_ok = True)

for i in range(1,10):
    created_file = open(os.path.join("NEKO", "neko{:03}.txt".format(i)) , "w")
    created_file.write(r"""neko{}です
neko
neko
neko
最終行です""".format(i))

    created_file.close()

#! python3
# check12_4.py  -  複数のテキストファイルをスプレッドシートに変換する


import os, openpyxl

wb = openpyxl.Workbook()
sheet = wb["Sheet"]


for i, txtfile in enumerate(os.listdir("NEKO")):
    f = open(os.path.join("NEKO", txtfile), "r")
    lines = f.readlines()

    for j in range(1, len(lines) + 1 ):
        sheet.cell(row = j, column = i+1).value = lines[j-1]
    
    f.close()

wb.save("ch12_4.xlsx")

#! python3
# check12_5.py  -  スプレッドシートを複数のテキストファイルに変換する


import os, openpyxl

wb = openpyxl.load_workbook("ch12_4.xlsx")
sheet = wb["Sheet"]


for i in range(1, sheet.max_column + 1):
    f = open(os.path.join("NEKO", "r_neko{:03}.txt".format(i)), "w")

    for j in range(1, sheet.max_row + 1):
        f.write(str(sheet.cell(row = j, column = i).value))
      
    f.close()