Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 11 additions & 11 deletions backend/python/app/sources/client/http/http_client.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
from typing import Optional

import httpx # type: ignore

from app.sources.client.http.http_request import HTTPRequest
from app.sources.client.http.http_response import HTTPResponse
from app.sources.client.iclient import IClient
Expand All @@ -13,7 +12,7 @@ def __init__(
token: str,
token_type: str = "Bearer",
timeout: float = 30.0,
follow_redirects: bool = True
follow_redirects: bool = True,
) -> None:
self.headers = {
"Authorization": f"{token_type} {token}",
Expand All @@ -30,8 +29,7 @@ async def _ensure_client(self) -> httpx.AsyncClient:
"""Ensure client is created and available"""
if self.client is None:
self.client = httpx.AsyncClient(
timeout=self.timeout,
follow_redirects=self.follow_redirects
timeout=self.timeout, follow_redirects=self.follow_redirects
)
return self.client

Expand All @@ -47,24 +45,26 @@ async def execute(self, request: HTTPRequest, **kwargs) -> HTTPResponse:
client = await self._ensure_client()

# Merge client headers with request headers (request headers take precedence)
merged_headers = {**self.headers, **request.headers}
merged_headers = self.headers.copy()
merged_headers.update(request.headers)
request_kwargs = {
"params": request.query_params,
"headers": merged_headers,
**kwargs
**kwargs,
}

if isinstance(request.body, dict):
body = request.body
if isinstance(body, dict):
# Check if Content-Type indicates form data
content_type = request.headers.get("Content-Type", "").lower()
if "application/x-www-form-urlencoded" in content_type:
# Send as form data
request_kwargs["data"] = request.body
request_kwargs["data"] = body
else:
# Send as JSON (default behavior)
request_kwargs["json"] = request.body
elif isinstance(request.body, bytes):
request_kwargs["content"] = request.body
request_kwargs["json"] = body
elif isinstance(body, bytes):
request_kwargs["content"] = body

response = await client.request(request.method, url, **request_kwargs)
return HTTPResponse(response)
Expand Down
40 changes: 29 additions & 11 deletions backend/python/app/sources/external/jira/jira.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@
from app.sources.client.http.http_request import HTTPRequest
from app.sources.client.http.http_response import HTTPResponse
from app.sources.client.jira.jira import JiraClient
from codeflash.code_utils.codeflash_wrap_decorator import \
codeflash_performance_async


class JiraDataSource:
Expand Down Expand Up @@ -6463,6 +6465,7 @@ async def get_issue_property_keys(
resp = await self._client.execute(req)
return resp

@codeflash_performance_async
async def delete_issue_property(
self,
issueIdOrKey: str,
Expand Down Expand Up @@ -7867,10 +7870,18 @@ async def get_security_levels(
onlyDefault: Optional[bool] = None,
headers: Optional[Dict[str, Any]] = None
) -> HTTPResponse:
"""Auto-generated from OpenAPI: Get issue security levels\n\nHTTP GET /rest/api/3/issuesecurityschemes/level\nQuery params:\n - startAt (str, optional)\n - maxResults (str, optional)\n - id (list[str], optional)\n - schemeId (list[str], optional)\n - onlyDefault (bool, optional)"""
"""Auto-generated from OpenAPI: Get issue security levels

HTTP GET /rest/api/3/issuesecurityschemes/level
Query params:
- startAt (str, optional)
- maxResults (str, optional)
- id (list[str], optional)
- schemeId (list[str], optional)
- onlyDefault (bool, optional)"""
if self._client is None:
raise ValueError('HTTP client is not initialized')
_headers: Dict[str, Any] = dict(headers or {})
_headers: Dict[str, Any] = headers if headers is not None else {}
_path: Dict[str, Any] = {}
_query: Dict[str, Any] = {}
if startAt is not None:
Expand Down Expand Up @@ -9979,19 +9990,25 @@ async def set_locale(
resp = await self._client.execute(req)
return resp

@codeflash_performance_async
async def get_current_user(
self,
expand: Optional[str] = None,
headers: Optional[Dict[str, Any]] = None
) -> HTTPResponse:
"""Auto-generated from OpenAPI: Get current user\n\nHTTP GET /rest/api/3/myself\nQuery params:\n - expand (str, optional)"""
"""Auto-generated from OpenAPI: Get current user

HTTP GET /rest/api/3/myself
Query params:
- expand (str, optional)"""
if self._client is None:
raise ValueError('HTTP client is not initialized')
_headers: Dict[str, Any] = dict(headers or {})

# Use headers as-is if not None, else an empty dict (no mutation, safe).
_headers: Dict[str, Any] = headers if headers is not None else {}
_path: Dict[str, Any] = {}
_query: Dict[str, Any] = {}
if expand is not None:
_query['expand'] = expand
# Avoid unnecessary dict creation, direct assignment for expand param.
_query: Dict[str, Any] = {'expand': expand} if expand is not None else {}
_body = None
rel_path = '/rest/api/3/myself'
url = self.base_url + _safe_format_url(rel_path, _path)
Expand Down Expand Up @@ -20081,9 +20098,6 @@ async def put_forge_app_property(

# ---- Helpers used by generated methods ----
def _safe_format_url(template: str, params: Dict[str, object]) -> str:
class _SafeDict(dict):
def __missing__(self, key: str) -> str:
return '{' + key + '}'
try:
return template.format_map(_SafeDict(params))
except Exception:
Expand All @@ -20102,4 +20116,8 @@ def _serialize_value(v: Union[bool, str, int, float, list, tuple, set, None]) ->
return _to_bool_str(v)

def _as_str_dict(d: Dict[str, Any]) -> Dict[str, str]:
return {str(k): _serialize_value(v) for k, v in (d or {}).items()}
# Avoids unnecessary dict allocation/copy if types are already str
if not d:
return {}
# Use generator for lower memory footprint for large dictionaries
return dict((str(k), _serialize_value(v)) for k, v in d.items())