import csv
import gzip

from .fields import QUERY_FIELDS


def get_query_result_rows(query_result_file):
    with gzip.open(query_result_file, mode="rt") as input_file:
        input_reader = csv.DictReader(
            input_file, fieldnames=QUERY_FIELDS, delimiter=",", quotechar='"'
        )

        yield from input_reader


def has_field_value(row, field):
    return row[field] != "\\\\N"