lines = open('app/__init__.py').readlines()
for i, line in enumerate(lines):
    if 'Content-Security-Policy' in line:
        indent = len(line) - len(line.lstrip())
        new_csp = (
            "\"default-src 'self'; "
            "script-src 'self' 'unsafe-inline' https://www.google.com/ https://www.gstatic.com/ https://recaptcha.google.com/ https://static.cloudflareinsights.com/; "
            "frame-src 'self' https://www.google.com/ https://recaptcha.google.com/; "
            "connect-src 'self' https://www.google.com/ https://recaptcha.google.com/ https://www.gstatic.com/; "
            "img-src 'self' data: https://www.gstatic.com/; "
            "style-src 'self' 'unsafe-inline';\""
        )
        lines[i] = ' ' * indent + "response.headers['Content-Security-Policy'] = " + new_csp + "\n"
        print("Updated line", i + 1)
open('app/__init__.py', 'w').writelines(lines)
print('Done - CSP updated')