Skip to content

Commit acaf44e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 835c43e commit acaf44e

File tree

3 files changed

+21
-18
lines changed

3 files changed

+21
-18
lines changed

config/settings/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -347,5 +347,5 @@
347347
LRM_DEV_PASSWORD = env("LRM_DEV_PASSWORD")
348348
LRM_QA_USER = env("LRM_QA_USER")
349349
LRM_QA_PASSWORD = env("LRM_QA_PASSWORD")
350-
LRM_DEV_TOKEN=env("LRM_DEV_TOKEN")
351-
XLI_TOKEN=env("XLI_TOKEN")
350+
LRM_DEV_TOKEN = env("LRM_DEV_TOKEN")
351+
XLI_TOKEN = env("XLI_TOKEN")

sde_collections/admin.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,14 +14,17 @@ def fetch_and_update_text_for_server(modeladmin, request, queryset, server_name)
1414
fetch_and_update_full_text.delay(collection.id, server_name)
1515
modeladmin.message_user(request, f"Started importing URLs from {server_name.upper()} Server")
1616

17+
1718
@admin.action(description="Import candidate URLs from LRM Dev Server with Full Text")
1819
def fetch_full_text_lrm_dev_action(modeladmin, request, queryset):
1920
fetch_and_update_text_for_server(modeladmin, request, queryset, "lrm_dev")
2021

22+
2123
@admin.action(description="Import candidate URLs from XLI Server with Full Text")
2224
def fetch_full_text_lis_action(modeladmin, request, queryset):
2325
fetch_and_update_text_for_server(modeladmin, request, queryset, "xli")
2426

27+
2528
@admin.action(description="Generate deployment message")
2629
def generate_deployment_message(modeladmin, request, queryset):
2730
# generate deployment message

sde_collections/sinequa_api.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
1-
from typing import Any
21
import json
2+
from typing import Any
3+
34
import requests
45
import urllib3
56
from django.conf import settings
@@ -100,26 +101,25 @@ def sql_query(self, sql: str) -> Any:
100101
raise ValueError("You must have a token to use the SQL endpoint")
101102

102103
url = f"{self.base_url}/api/v1/engine.sql"
103-
headers = {
104-
"Content-Type": "application/json",
105-
"Authorization": f"Bearer {self.token}"
106-
}
107-
payload = json.dumps({
108-
"method": "engine.sql",
109-
"sql": sql,
110-
"pretty": True,
111-
"log": False,
112-
"output": "json",
113-
"resolveIndexList": "false",
114-
"engines": "default",
115-
})
104+
headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.token}"}
105+
payload = json.dumps(
106+
{
107+
"method": "engine.sql",
108+
"sql": sql,
109+
"pretty": True,
110+
"log": False,
111+
"output": "json",
112+
"resolveIndexList": "false",
113+
"engines": "default",
114+
}
115+
)
116116
try:
117117
response = requests.post(url, headers=headers, data=payload, timeout=10)
118118
response.raise_for_status()
119119
return response.json()
120120
except requests.exceptions.RequestException as e:
121121
raise Exception(f"API request failed: {str(e)}")
122-
122+
123123
def get_full_texts(self, collection_config_folder: str) -> Any:
124124
sql = f"SELECT url1, text, title FROM sde_index WHERE collection = '/SDE/{collection_config_folder}/'"
125-
return self.sql_query(sql)
125+
return self.sql_query(sql)

0 commit comments

Comments
 (0)