#!/usr/bin/env python3
"""Fix _cfg normalization in search_app.py"""
with open('/root/search_app.py', 'r') as f:
    c = f.read()

old_line = "                        _cfg = _j.load(open('/root/gov_crawler/daily_crawl_config.json'))"
new_lines = ("                        _cfg = _j.load(open('/root/gov_crawler/daily_crawl_config.json'))\n"
             "                        if isinstance(_cfg, list):\n"
             "                            _cfg = {'crawlers': _cfg}")

if old_line in c:
    c = c.replace(old_line, new_lines, 1)
    with open('/root/search_app.py', 'w') as f:
        f.write(c)
    print("search_app.py: patched")
else:
    print("search_app.py: line not found!")
    idx = c.find('_cfg = _j.load')
    if idx >= 0:
        print("Found at:", idx)
        print(repr(c[idx:idx+200]))