wip

bayer/views.py

@@ -1,13 +1,41 @@
+import base64
+
 import requests
 from django.conf import settings as s
+from django.core.exceptions import BadRequest
 from django.http import HttpResponse
-from pydantic import BaseModel
+from django.shortcuts import redirect
 
-from utilities.chem import FormatConverter
+from bayer.models import PESCompound
+from epdb.logic import PackageManager
+from epdb.views import _anonymous_or_real
+from utilities.decorators import package_permission_required
 
 
-class PES(BaseModel):
-    pass
+@package_permission_required()
+def create_pes(request, package_uuid):
+    current_user = _anonymous_or_real(request)
+    current_package = PackageManager.get_package_by_id(current_user, package_uuid)
+
+    if request.method == "POST":
+        compound_name = request.POST.get('compound-name')
+        compound_description = request.POST.get('compound-description')
+        pes_link = request.POST.get('pes-link')
+
+        if pes_link:
+            try:
+                pes_data = fetch_pes(request, pes_link)
+            except ValueError as e:
+                return BadRequest(f"Could not fetch PES data for {pes_link}")
+
+            pes = PESCompound.create(current_package, pes_data, compound_name, compound_description)
+
+            return redirect(pes.url)
+        else:
+            return BadRequest("Please provide a PES link.")
+    else:
+        pass
+
 
 
 def fetch_pes(request, pes_url) -> dict:
@@ -19,28 +47,35 @@ def fetch_pes(request, pes_url) -> dict:
     from epauth.views import get_access_token_from_request
     token = get_access_token_from_request(request)
 
-    if token:
+    if token or True:
         for k, v in s.PES_API_MAPPING.items():
-            if pes_url.startsWith(k):
+            if pes_url.startswith(k):
                 pes_id = pes_url.split('/')[-1]
-                headers = {"Authorization": f"Bearer {token['access_token']}"}
-                params = {"pes_reg_entity_corporate_id": pes_id}
 
-                res = requests.get(v, headers=headers, params=params, proxies=proxies)
-
-                try:
-                    res.raise_for_status()
-                    pes_data = res.json()
-
-                    if len(pes_data) == 0:
-                        raise ValueError(f"PES with id {pes_id} not found")
-
-                    res_data = pes_data[0]
+                if pes_id == 'dummy' or True:
+                    import json
+                    res_data = json.load(open(s.BASE_DIR / "fixtures/pes.json"))
                     res_data["pes_url"] = pes_url
                     return res_data
+                else:
+                    headers = {"Authorization": f"Bearer {token['access_token']}"}
+                    params = {"pes_reg_entity_corporate_id": pes_id}
 
-                except requests.exceptions.HTTPError as e:
-                    raise ValueError(f"Error fetching PES with id {pes_id}: {e}")
+                    res = requests.get(v, headers=headers, params=params, proxies=proxies)
+
+                    try:
+                        res.raise_for_status()
+                        pes_data = res.json()
+
+                        if len(pes_data) == 0:
+                            raise ValueError(f"PES with id {pes_id} not found")
+
+                        res_data = pes_data[0]
+                        res_data["pes_url"] = pes_url
+                        return res_data
+
+                    except requests.exceptions.HTTPError as e:
+                        raise ValueError(f"Error fetching PES with id {pes_id}: {e}")
         else:
             raise ValueError(f"Unknown URL {pes_url}")
     else:
@@ -52,8 +87,10 @@ def visualize_pes(request):
 
     if pes_link:
         pes_data = fetch_pes(request, pes_link)
-        print(pes_data)
 
-    return HttpResponse(
-        FormatConverter.to_png("c1ccccc1"), content_type="image/png"
-    )
+        representations = pes_data.get('representations')
+
+        for rep in representations:
+            if rep.get('type') == 'color':
+                image_data = base64.b64decode(rep.get('base64').replace("data:image/png;base64,", ""))
+                return HttpResponse(image_data, content_type="image/png")