def choose_writer_class()

in psaw/psaw.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

def choose_writer_class(format, batch_mode):
    """
    Choose appropriate writer class

    :param format: str
    :param batch_mode: bool
    :return: Class

    """
    writer_cls = {
        ('json', False): wt.JsonWriter,
        ('json', True): wt.JsonBatchWriter,
        ('csv', False): wt.CsvWriter,
        ('csv', True): wt.CsvBatchWriter,
    }[(format, batch_mode)]

    return writer_cls