def test_create_workbook_default_sheet(fmt):
library = Files()
library.create_workbook(fmt=fmt)
assert library.list_worksheets() == ["Sheet"]
library.create_worksheet("Test")
assert library.list_worksheets() == ["Sheet", "Test"]
def generate_data(user_count=100, feedback_count=5):
fake = Faker()
files = Files()
files.create_workbook("userdata.xlsx")
files.create_worksheet("profile", exist_ok=True)
files.create_worksheet("feedback", exist_ok=True)
users = []
names = []
for i in range(user_count):
userprofile = fake.profile()
if userprofile["name"] in names:
print("name %s already exists", userprofile["name"])
continue
userprofile.pop("website")
userprofile.pop("current_location")
files.append_rows_to_worksheet(userprofile, "profile", header=True)
users.append(userprofile)
for i in range(1, 16):
selected_user = users[i]
feedback = {
"name": selected_user["name"],
"feedback": choice(good_feedback)
}
print(feedback)
files.append_rows_to_worksheet(feedback, "feedback", header=True)
for i in range(17, 22):
selected_user = users[i]
feedback = {
"name": selected_user["name"],
"feedback": choice(neutral_feedback)
}
print(feedback)
files.append_rows_to_worksheet(feedback, "feedback", header=True)
for i in range(23, 28):
selected_user = users[i]
feedback = {
"name": selected_user["name"],
"feedback": choice(bad_feedback)
}
print(feedback)
files.append_rows_to_worksheet(feedback, "feedback", header=True)
files.remove_worksheet("Sheet")
files.save_workbook()
