Maks1mS/ocab
0
0
Fork 0
mirror of https://gitflic.ru/project/maks1ms/ocab.git synced 2024-12-23 16:23:01 +03:00
Code

Добавлены nosec для прохождения bandit

Browse Source
This commit is contained in:
Maxim Slipenko 2024-07-08 00:55:33 +03:00
parent ef0dda07f7
commit ef10f05a73
5 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/modules/external/yandexgpt/yandexgpt.py vendored
View File

@ -99,7 +99,7 @@ class YandexGPT:
"messages": messages, "messages": messages,
} }
response = requests.post(url, headers=headers, json=prompt).text response = requests.post(url, headers=headers, json=prompt).text # nosec
return json.loads(response)["result"]["alternatives"][0]["message"]["text"] return json.loads(response)["result"]["alternatives"][0]["message"]["text"]
async def async_yandexgpt( async def async_yandexgpt(
@ -135,7 +135,9 @@ class YandexGPT:
}, },
"messages": messages, "messages": messages,
} }
response = await self.async_request(url=url, headers=headers, prompt=request) response = await self.async_request(
url=url, headers=headers, prompt=request
) # nosec
return response["result"]["alternatives"][0]["message"]["text"] return response["result"]["alternatives"][0]["message"]["text"]
async def async_yandexgpt_translate(self, input_language, output_language, text): async def async_yandexgpt_translate(self, input_language, output_language, text):
@ -187,7 +189,7 @@ class YandexGPT:
"messages": messages, "messages": messages,
} }
response = requests.post(url, headers=headers, json=prompt).text response = requests.post(url, headers=headers, json=prompt).text # nosec
return json.loads(response)["result"]["alternatives"][0]["message"]["text"] return json.loads(response)["result"]["alternatives"][0]["message"]["text"]
async def async_yandex_cloud_text_to_speech( async def async_yandex_cloud_text_to_speech(

2
src/modules/standard/config/tests/test_config.py
View File

@ -33,7 +33,7 @@ class TestConfig(unittest.TestCase):
) )
def test_yaml_values(self): def test_yaml_values(self):
expected_token = "xxxxxxxxxxxxxxxxxxxx" expected_token = "xxxxxxxxxxxxxxxxxxxx" # nosec
expected_role_values = {"ADMIN": 0, "MODERATOR": 1, "USER": 2, "BOT": 3} expected_role_values = {"ADMIN": 0, "MODERATOR": 1, "USER": 2, "BOT": 3}
self.assertEqual(yaml_load["TELEGRAM"]["TOKEN"], expected_token) self.assertEqual(yaml_load["TELEGRAM"]["TOKEN"], expected_token)

2
src/modules/standard/database/tests/test_db.py
View File

@ -135,7 +135,7 @@ class TestDatabaseAPI(unittest.TestCase):
@classmethod @classmethod
def tearDownClass(cls): def tearDownClass(cls):
cls.database.close() cls.database.close()
os.system(f"rm {cls.path}") os.system(f"rm {cls.path}") # nosec
if __name__ == "__main__": if __name__ == "__main__":

2
src/modules/standard/roles/tests/test_roles.py
View File

@ -73,7 +73,7 @@ class TestRoles(unittest.IsolatedAsyncioTestCase):
@classmethod @classmethod
def tearDownClass(cls): def tearDownClass(cls):
cls.database.close() cls.database.close()
os.system(f"rm {cls.path}") os.system(f"rm {cls.path}") # nosec
if __name__ == "__main__": if __name__ == "__main__":

8
src/modules/standard/welcome/handlers.py
View File

@ -16,13 +16,13 @@ from src.modules.standard.roles.roles import Roles
async def create_math_task(): async def create_math_task():
first_number = random.randint(1, 100) first_number = random.randint(1, 100) # nosec
second_number = random.randint(1, 100) second_number = random.randint(1, 100) # nosec
answer = first_number + second_number answer = first_number + second_number
fake_answers = [] fake_answers = []
for i in range(3): for i in range(3):
diff = random.randint(1, 10) diff = random.randint(1, 10) # nosec
diff_sign = random.choice(["+", "-"]) diff_sign = random.choice(["+", "-"]) # nosec
fake_answers.append(answer + diff if diff_sign == "+" else answer - diff) fake_answers.append(answer + diff if diff_sign == "+" else answer - diff)
fake_answers.append(answer) fake_answers.append(answer)
random.shuffle(fake_answers) random.shuffle(fake_answers)