Python - несколько экземпляров в цикле - PullRequest
Новая
       15

Python - несколько экземпляров в цикле

2 голосов
/ 03 марта 2020

У меня есть следующий класс для генерации случайных простых файлов PDF. 

from reportlab.pdfgen.canvas import Canvas
from reportlab.lib.pagesizes import letter
import random
import string
import os

BOOTH_NAME = [
        'Distrito', 'Sector', 'Residencia', 'Fraccionamiento',
        'Privado', 'Ciudad', 'Colonia', 'Departamentos',
        'Recinto', 'Barrio', 'Comuna', 'Vecindad'
    ]
BOOTH_CODE = list(string.ascii_uppercase)
ZONE_CODE = ['NORTE', 'OESTE', 'SUR', 'ESTE', 'SUROESTE', 'NOROESTE', 'SURESTE', 'NORESTE']
CONCEPT_NAME = ['COMPRA/VENTA', 'TESTIMONIO', 'ESCRITURAS']
WIDTH, HEIGHT = letter
DIR_NAME = '/Users/gmwill934/PycharmProjects/notarIA/pdfs/'

class PDF(object):
    def __init__(
            self,
            casilla=str(BOOTH_NAME[random.randint(0, len(BOOTH_NAME) - 1)]),
            clave_casilla=str(BOOTH_CODE[random.randint(0, len(BOOTH_CODE) - 1)]),
            zona=str(ZONE_CODE[random.randint(0, len(ZONE_CODE) - 1)]),
            concepto=str(CONCEPT_NAME[random.randint(0, len(CONCEPT_NAME) - 1)])
    ):
        self.casilla = casilla
        self.clave_casilla = clave_casilla
        self.zona = zona
        self.concepto = concepto
        self.name = '{} {} {}'.format(self.casilla, self.clave_casilla, self.zona)
        self.save_name = os.path.join(PDF.DIR_NAME, self.name+'.pdf')

    def create_pdf(self):
        pdf = Canvas(self.save_name, pagesize=letter)
        pdf.setTitle(self.name)
        pdf.grid([10, PDF.WIDTH - 10], [10, PDF.HEIGHT - 10])
        pdf.drawString(20, PDF.HEIGHT - 30, 'Numero de Casilla:')
        pdf.drawString(130, PDF.HEIGHT - 30, str(random.randint(1000, 9999)))
        pdf.drawString(20, PDF.HEIGHT - 50, 'Nombre de Casilla:')
        pdf.drawString(130, PDF.HEIGHT - 50, self.name)
        pdf.drawString(20, PDF.HEIGHT - 70, 'Numero de Votos:')
        pdf.drawString(130, PDF.HEIGHT - 70, str(random.randint(100000, 999999)))
        pdf.drawString(20, PDF.HEIGHT - 90, 'Concepto:')
        pdf.drawString(130, PDF.HEIGHT - 90, str(self.concepto))
        pdf.showPage()
        pdf.save()

В основном метод create_pdf создает файлы PDF со случайными значениями.

Что я хочу сделать дальше, это создать несколько экземпляров PDF class, для создания нескольких PDF-файлов одновременно. 

from src.pdf import PDF
# instantiate PDF class
# create multiple instances
pdf = PDF()
for n in range(10):
    pdf.create_pdf()

Также я попробовал это. 

from src.pdf import PDF
# instantiate PDF class
# create multiple instances
for n in range(10):
    pdf = PDF()
    pdf.create_pdf()

Проблема в том, что он создает только 1 файл PDF, в то время как Я ожидаю 10. Кажется, что он закончен после создания 1 PDF.

Может кто-нибудь посоветовать по этому поводу? Я что-то упустил?

1 Ответ

3 голосов
/ 03 марта 2020

Значение по умолчанию для параметров, оцениваемых, когда функция (метод) определена .
Таким образом, ваш случайный генерирующий лог c запускается только один раз.
Вы должны сделать: 

    def __init__(
            self,
            casilla=None,
            ...
    ):
        self.casilla = (
            casilla if casilla is not None else
            str(BOOTH_NAME[random.randint(0, len(BOOTH_NAME) - 1)]))
        ...

ссылка: https://docs.python.org/3/reference/compound_stmts.html#function -определения

Значения параметров по умолчанию оцениваются слева направо при выполнении определения функции. Это означает, что Выражение вычисляется один раз, когда функция определена, и для каждого вызова используется одно и то же «предварительно вычисленное» значение. Это особенно важно понимать, когда параметр по умолчанию является изменяемым объектом, таким как список или словарь: если функция изменяет объект (например, путем добавления элемента в список), значение по умолчанию фактически изменяется. Это вообще не то, что было задумано. Чтобы обойти это, используйте None по умолчанию и явно проверьте его в теле функции, например: 

def whats_on_the_telly(penguin=None):
    if penguin is None:
        penguin = []
    penguin.append("property of the zoo")
    return penguin
...