From 909c95ed7a1748cb32d2e5eabc9a13b461a79783 Mon Sep 17 00:00:00 2001
From: TPD94 <>
Date: Sat, 27 Apr 2024 18:05:35 -0400
Subject: [PATCH] Added proxy support

- Minor code change to prevent bug
---
 decrypt.py | 2 +-
 main.py    | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/decrypt.py b/decrypt.py
index 2e5c7a9..6a681cd 100644
--- a/decrypt.py
+++ b/decrypt.py
@@ -36,7 +36,7 @@ def decrypt_content(in_pssh: str = None, license_url: str = None, headers: dict
     # Generate the challenge
     challenge = cdm.get_license_challenge(session_id, pssh)
 
-    if proxy:
+    if proxy is not None:
         # send license challenge
         license = requests.post(
             url=license_url,
diff --git a/main.py b/main.py
index e87ffaa..2caa85e 100644
--- a/main.py
+++ b/main.py
@@ -31,6 +31,8 @@ def post_endpoint():
     if data['proxy'] != None:
         proxy = data['proxy']
         print(proxy)
+    else:
+        proxy = None
 
     # Format the headers