Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Sweep: Write a reverse proxy that edit the api_token only (✓ Sandbox Passed) #91

Closed
wants to merge 2 commits into from
Closed

Sweep: Write a reverse proxy that edit the api_token only (✓ Sandbox Passed) #91

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
23fb3f3
feat: Add token_handler module to modify api_token
onuratakan Mar 13, 2024
e2100e3
feat: Updated upsonic_on_prem/api/pre_process/base
onuratakan Mar 13, 2024
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
4 changes: 4 additions & 0 deletions upsonic_on_prem/api/pre_process/base.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,13 @@

from upsonic_on_prem.api.pre_process.user import *

from upsonic_on_prem.api.pre_process.token_handler import edit_api_token_if_needed


@app.before_request
def check():
edit_api_token_if_needed(request)

the_endpoint = request.endpoint
if request.endpoint == None:
the_endpoint = ""
Expand Down
13 changes: 13 additions & 0 deletions upsonic_on_prem/api/pre_process/token_handler.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
from flask import request

Check warning on line 1 in upsonic_on_prem/api/pre_process/token_handler.py

View workflow job for this annotation

GitHub Actions / Qodana Community for Python

Unsatisfied package requirements 

Package requirements 'fire==0.5.0', 'python-dotenv==1.0.0', 'flask==2.0.0', 'waitress==2.1.2', 'Flask-Limiter==3.4.0', 'werkzeug==2.3.7', 'redis==4.6.0', 'rich==13.7.0', 'cloudpickle==3.0.0', 'requests==2.28.2', 'cryptography==41.0.3', 'django==4.2.9', 'gunicorn==20.1.0', 'tzdata==2023.3', 'sentry-sdk\[flask\]==1.40.0', 'sentry-sdk\[django\]==1.40.0', 'pytest==8.0.0', 'dill==0.3.7', 'django-allauth==0.58.0', 'django-crispy-forms==2.1', 'crispy-bootstrap5==0.7', 'logtail-python==0.2.10', 'ollama==0.1.6', 'langchain==0.1.10', 'langchain-community==0.0.25', 'chromadb==0.4.24', 'openai==1.12.0', 'django-pwa==1.1.0' are not satisfied


def modify_token(original_token):
# Placeholder logic for token modification
return f"{original_token}-modified"

def edit_api_token_if_needed(req):

Check notice on line 8 in upsonic_on_prem/api/pre_process/token_handler.py

View workflow job for this annotation

GitHub Actions / Qodana Community for Python

PEP 8 coding style violation 

PEP 8: E302 expected 2 blank lines, found 1
api_token = req.headers.get('api_token')
if api_token:
modified_token = modify_token(api_token)
# Demonstrating token modification. In a real scenario, this might involve more complex logic or direct request manipulation.

Check notice on line 12 in upsonic_on_prem/api/pre_process/token_handler.py

View workflow job for this annotation

GitHub Actions / Qodana Community for Python

PEP 8 coding style violation 

PEP 8: E501 line too long (133 \> 120 characters)
req.headers['api_token'] = modified_token
Loading