blank_row_inserter.py

#!/usr/bin/env python3

"""Replicate a spreadsheet in CWD but with N blank rows inserted at chosen place.

Usage: blank_row_inserter.py takes 3 arguments
<Name> - Name of the spreadsheet to add blank rows to
<Start> - The row to start inserting the blank lines
<Number> - The number of blank lines to insert

e.g. python blank_row_inserter.py test.xlsx 3 2
     Would Insert 2 blank rows starting at row 3 into a copy of 'test.xlxs'
"""

import sys
import openpyxl

name = sys.argv[1]
blank_start = int(sys.argv[2])
blank_length = int(sys.argv[3])

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

# Read the data from spreadsheet and get list of row data lists
print('Reading spreadsheet data...')
rows = []
max_row = sheet.max_row
max_column = sheet.max_column
for row in range(1, max_row + 1):
    data = []
    for cell in range(1, max_column + 1):
        cell_value = sheet.cell(row=row, column=cell).value
        data.append(cell_value)
    rows.append(data)

#print(sheet_data)

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

# Write rows prior to into new spreadsheet with blank lines inserted
print('Inserting blanks...')
for row in range(1, blank_start):
    for cell in range(1, max_column + 1):
        sheet.cell(row=row, column=cell).value = rows[row - 1][cell - 1]

# Write remaining rows after the blank gap
for row in range(blank_start + blank_length, max_row + blank_length + 1):
    for cell in range(1, max_column + 1):
        sheet.cell(row=row, column=cell).value = (rows[row - blank_length - 1]
                                                  [cell - 1])

wb.save('blanked-' + name)

print("A copy of the spreadsheet with blanks inserted has been saved"
      " as 'blanked-spreadsheet_name]'. It can be found in the same"
      " directory as the original.")