Merge pull request #46 from azuax/main

Read HEADERS configuration from config.py
dolevf · Sep 30, 2024 · c29656d · c29656d
2 parents 701e4c1 + 5cd734e
commit c29656d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -87,6 +87,11 @@ def main():
     for header in options.header:
       key, value = header.split(": ")
       headers[key] = value
+
+    # Read configuration headers
+    for header in conf.HEADERS:
+        headers[header] = conf.HEADERS[header]
+
     g = GRAPHW00F(follow_redirects=options.followredirect,
                   headers=headers,
                   cookies=conf.COOKIES,
@@ -170,4 +175,4 @@ def main():
 
 if __name__ == '__main__':
     main()
-
+