codespector 0.2.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

codespector/codespector.py

@@ -18,6 +18,7 @@ class CodeSpector:
         result_file: str,
         output_dir: str,
         chat_model: str | None = None,
+        exclude_file_ext: list[str] | None = None,
     ):
         self.chat_token = chat_token
         self.chat_agent = chat_agent
@@ -30,6 +31,7 @@ class CodeSpector:
         self.pipeline = pipeline
         self.result_file = result_file
         self.chat_model = chat_model
+        self.exclude_file_exc: list[str] | None = (None,)
 
     @classmethod
     def create(
@@ -42,6 +44,7 @@ class CodeSpector:
         result_file: str,
         output_dir: str,
         chat_model: str | None = None,
+        exclude_file_ext: list[str] | None = None,
     ) -> 'CodeSpector':
         agent_info = AgentInfo.create(
             chat_agent=chat_agent,
@@ -51,6 +54,7 @@ class CodeSpector:
         data_preparer = CodeSpectorDataPreparer(
             output_dir=output_dir,
             compare_branch=compare_branch,
+            exclude_file_ext=exclude_file_ext,
         )
         reviewer = CodeSpectorReviewer(
             diff_file=data_preparer.combined_file,
@@ -77,6 +81,7 @@ class CodeSpector:
             pipeline=pipeline,
             output_dir=output_dir,
             result_file=result_file,
+            exclude_file_ext=exclude_file_ext,
         )
 
     def review(self):

codespector/local.py

@@ -10,9 +10,11 @@ class CodeSpectorDataPreparer(BasePipe):
         self,
         output_dir: str,
         compare_branch: str,
+        exclude_file_ext: list[str] | None = None,
     ):
         self.output_dir = output_dir
         self.compare_branch = compare_branch
+        self.exclude_file_ext = exclude_file_ext
 
         self.original_files_tmp = 'original_files_tmp.json'
         self.code_changes_only = 'code_changes_only.txt'
@@ -61,8 +63,9 @@ class CodeSpectorDataPreparer(BasePipe):
         result = {'original files': []}
 
         for file in changed_files:
-            if not file.endswith('.py'):
-                continue
+            if self.exclude_file_ext:
+                if file.endswith(tuple(self.exclude_file_ext)):
+                    continue
 
             if os.path.isfile(file):
                 with open(file, 'r', encoding='utf-8') as f:

codespector/main.py

@@ -11,6 +11,13 @@ env = Env()
 env.read_env(path=str(BASE_PATH / '.env'))
 
 
+@click.option(
+    '--exclude-file-ext',
+    type=list,
+    envvar='CODESPECTOR_EXCLUDE_FILE_EXT',
+    help='Exclude file extensions from the review',
+    show_envvar=True,
+)
 @click.option(
     '--result-file',
     type=str,
@@ -50,7 +57,7 @@ env.read_env(path=str(BASE_PATH / '.env'))
     '--chat-agent',
     envvar='CODESPECTOR_CHAT_AGENT',
     show_envvar=True,
-    help='Choose the chat agent to use you can use one from [codestral, chatgpr, deepseek]. Or set yours chat agent',
+    help='Choose the chat agent to use you can use one from [codestral, chatgpt, deepseek]. Or set yours chat agent',
 )
 @click.option(
     '--chat-model',

{codespector-0.2.0.dist-info → codespector-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codespector
-Version: 0.2.0
+Version: 1.0.1
 Summary: Assistant for reviewing your code
 Project-URL: Repository, https://github.com/Vladimir-Titov/codespector
 Project-URL: Issues, https://github.com/Vladimir-Titov/codespector/issues
@@ -50,18 +50,23 @@ You can use the `codespector` command to start a code review. Below are the avai
 Usage: codespector [OPTIONS]
 
 Options:
-  --system-content TEXT       Content which used in system field for agent
-                              [default: Ты код ревьювер. Отвечай на русском языке.]
-  --output-dir TEXT           Select the output directory [default: codespector]
-  -b, --compare-branch TEXT   Select the branch to compare the current one with
-                              [default: develop]
-  --chat-agent [codestral|chatgpt]
-                              Choose the chat agent to use [default: codestral]
-  --chat-model TEXT           Choose the chat model to use
-  --chat-token TEXT           Chat agent token
-  --mode [local]              Choose the mode of the application [default: local]
-  --version                   Show the version and exit.
-  --help                      Show this message and exit.
+  --chat-token TEXT          Chat agent token  [env var: CODESPECTOR_CHAT_TOKEN]
+  --chat-model TEXT          Choose the chat model to use  [env var: CODESPECTOR_CHAT_MODEL]
+  --chat-agent TEXT          Choose the chat agent to use (codestral, chatgpt, deepseek)
+                            or set your own  [env var: CODESPECTOR_CHAT_AGENT]
+  -b, --compare-branch TEXT  Select the branch to compare the current one with
+  --output-dir TEXT         Select the output directory  [default: codespector]
+                            [env var: CODESPECTOR_OUTPUT_DIR]
+  --system-content TEXT     Content which used in system field for agent
+                            [env var: CODESPECTOR_SYSTEM_CONTENT]
+  --prompt-content TEXT     Prompt content which included to review prompt
+                            [env var: CODESPECTOR_PROMPT_CONTENT]
+  --result-file TEXT        Set file for saving the result
+                            [env var: CODESPECTOR_RESULT_FILE]
+  --exclude-file-ext LIST   Exclude file extensions from the review
+                            [env var: CODESPECTOR_EXCLUDE_FILE_EXT]
+  --version                 Show the version and exit.
+  --help                    Show this message and exit.
 ```
 
 ### Example
@@ -69,7 +74,7 @@ Options:
 To run a code review, use the following command:
 
 ```sh
-codespector --chat-token YOUR_CHAT_TOKEN --chat-agent codestral --compare-branch develop
+codespector --chat-token YOUR_CHAT_TOKEN --chat-agent codestral --compare-branch develop --result-file result.md --system-content "system content" --prompt-content "prompt content"
 ```
 
 ## Configuration
@@ -77,11 +82,14 @@ codespector --chat-token YOUR_CHAT_TOKEN --chat-agent codestral --compare-branch
 You can also configure CodeSpector using environment variables. Create a `.env` file in the root directory of your project with the following content:
 
 ```
-CODESPECTOR_SYSTEM_CONTENT=Ты код ревьювер. Отвечай на русском языке.
+CODESPECTOR_CHAT_TOKEN=your_token
 CODESPECTOR_OUTPUT_DIR=codespector
+CODESPECTOR_SYSTEM_CONTENT="Ты код ревьювер. Отвечай на русском"
+CODESPECTOR_PROMPT_CONTENT="Оцени код на безопасноть, соблюдение лучших техник"
+CODESPECTOR_RESULT_FILE="result.md"
 CODESPECTOR_CHAT_AGENT=codestral
-CODESPECTOR_CHAT_MODEL=codestral-latest
-CODESPECTOR_CHAT_TOKEN=YOUR_CHAT_TOKEN
+CODESPECTOR_CHAT_MODEL=your_model
+CODESPECTOR_EXCLUDE_FILE_EXT=.pyc,.pyo
 ```
 
 ## Makefile Commands

codespector-1.0.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+codespector/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+codespector/base.py,sha256=uPoJ8h0Dlna-eUXyyD9s7uaQ4z0pS6qPdQMeu10muEk,159
+codespector/codespector.py,sha256=XHUbdWOZB0HSpXLdpDDFDIse9tiwVJ-gSyOPV0SCHcw,2842
+codespector/errors.py,sha256=Z5W6i7iG_iBj_IfdSZ_f9jij0oPxTZ2Cxs87kzk3oyU,377
+codespector/local.py,sha256=neODRe9DwQ_ghXccXVc-A-hdVA6Tz7EtVcmCwUh4HHs,2965
+codespector/main.py,sha256=NoXN8nraW6_EtDK1oOx63rojs_fuSNeM2VX-rsSyg7o,1882
+codespector/reviewer.py,sha256=WoKRO-bFFLKNUiWB_TWoK0rRjenHkJ5fej_lhcj_e2Q,2740
+codespector/types.py,sha256=i5PFazyUc5rVq5Fhn4_FsG7R1l617vyjByvnixDUw1Y,1255
+codespector-1.0.1.dist-info/METADATA,sha256=j5kdhhqRBlgetvHhABTUriuVenzHVx1CnXyblK8CPQI,3791
+codespector-1.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+codespector-1.0.1.dist-info/entry_points.txt,sha256=QlHn96KY8vzY1sOweKIuZOAQrSVse6h3v57vkwmHmJg,54
+codespector-1.0.1.dist-info/licenses/LICENSE,sha256=Eta34ENUL_dZWy-prVNucMkoA38WIqiO9pKTUYiuT_A,1070
+codespector-1.0.1.dist-info/RECORD,,

codespector-0.2.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-codespector/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-codespector/base.py,sha256=uPoJ8h0Dlna-eUXyyD9s7uaQ4z0pS6qPdQMeu10muEk,159
-codespector/codespector.py,sha256=qW9G3arSq3VzWxdvPP4uCMNFfWFd7HBqs8M8dQsOAHU,2588
-codespector/errors.py,sha256=Z5W6i7iG_iBj_IfdSZ_f9jij0oPxTZ2Cxs87kzk3oyU,377
-codespector/local.py,sha256=aEq1YBm4fAFywpwmTaAdpM0_C2RHuIFDmAKikUZkmtQ,2800
-codespector/main.py,sha256=dfCds8mEIVb7mHVoczijnceIOd94I9yJc-8ZWaXcllo,1707
-codespector/reviewer.py,sha256=WoKRO-bFFLKNUiWB_TWoK0rRjenHkJ5fej_lhcj_e2Q,2740
-codespector/types.py,sha256=i5PFazyUc5rVq5Fhn4_FsG7R1l617vyjByvnixDUw1Y,1255
-codespector-0.2.0.dist-info/METADATA,sha256=YNuCT3bCxjyw46_gFGQdqwTwTOhswYWMqFRTF7eFXD0,3131
-codespector-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-codespector-0.2.0.dist-info/entry_points.txt,sha256=QlHn96KY8vzY1sOweKIuZOAQrSVse6h3v57vkwmHmJg,54
-codespector-0.2.0.dist-info/licenses/LICENSE,sha256=Eta34ENUL_dZWy-prVNucMkoA38WIqiO9pKTUYiuT_A,1070
-codespector-0.2.0.dist-info/RECORD,,