I want to reduce the code repetition by having to create a variable with a DictWriter and all he information every time I need to write in a csv file.

class Example:

    def func1():
        with open('filepath', 'w') as file:
            csv_writer = DictWriter(file, fieldnames=fieldnames, lineterminator='\n')

            ...

    def func2():
        with open('filepath', 'a') as file:
            csv_writer = DictWriter(file, fieldnames=fieldnames, lineterminator='\n')

            ...

I thinked in create a private class atribute using the open function to pass the file, like _csv_writer = DictWriter(open('filepath')) to don’t need to create a DictWriter everytime, but if i do that the file will not close like in the with manager right? So have any other option to reduce that code and still closing the file?

>Solution :

You can’t easily avoid the open() calls because they’re in a context manager (unless you want to define your own context manager), but you can define a function that calls DictWriter() with your options so you don’t have to repeat them.

def myDictWriter(file):
    return DictWriter(file, fieldnames=fieldnames, lineterminator='\n')

class Example:

    def func1():
        with open('filepath', 'w') as file:
            csv_writer = myDictWriter(file)

            ...

    def func2():
        with open('filepath', 'w') as file:
            csv_writer = myDictWriter(file)