ad_source.py

ad_source_id = 0

cs = [('ad_image_domain', imp['ac_bannerhost']),
      ('iframe_domain', data_rec['iframe_domain']),
      ('ad_image_linked_domain', imp['ac_clickhost'])]
cs += [('iframe_domain', iframe_dom)
       for iframe_dom, _, _ in iframe_stack]

for company_table, company_value in cs:
    q = """select company from {}
           where data = %s
        """.format(company_table)
    company = conn.execute(q, company_value).scalar()
    if not company:
        continue
    q = """select id from ad_source
           where data = %s
        """
    ad_source_id = conn.execute(q, company).scalar()
    if ad_source_id:
        break

if ad_source_id:
    data_rec['ad_source'] = ad_source_id
else:
    q = """select id from ad_source
           where data = 'Media Buy'
        """
    data_rec['ad_source'] = conn.execute(q).scalar() or 0