Skip to content

Commit cbd2cf4

Browse files
committed
linting
1 parent accfe2f commit cbd2cf4

File tree

4 files changed

+46
-28
lines changed

4 files changed

+46
-28
lines changed

rsconnect/environment.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,11 @@ def create_python_environment(
161161

162162
@classmethod
163163
def _get_python_env_info(
164-
cls, file_name: str, python: typing.Optional[str], force_generate: bool = False, require_requirements_txt: bool = True
164+
cls,
165+
file_name: str,
166+
python: typing.Optional[str],
167+
force_generate: bool = False,
168+
require_requirements_txt: bool = True,
165169
) -> "Environment":
166170
"""
167171
Gathers the python and environment information relating to the specified file
@@ -176,7 +180,12 @@ def _get_python_env_info(
176180
"""
177181
python = which_python(python)
178182
logger.debug("Python: %s" % python)
179-
environment = cls._inspect_environment(python, os.path.dirname(file_name), force_generate=force_generate, require_requirements_txt=require_requirements_txt)
183+
environment = cls._inspect_environment(
184+
python,
185+
os.path.dirname(file_name),
186+
force_generate=force_generate,
187+
require_requirements_txt=require_requirements_txt,
188+
)
180189
if environment.error:
181190
raise RSConnectException(environment.error)
182191
logger.debug("Python: %s" % python)
@@ -204,10 +213,10 @@ def _inspect_environment(
204213
args = [python, "-m", "rsconnect.subprocesses.inspect_environment"]
205214
if flags:
206215
args.append("-" + "".join(flags))
207-
216+
208217
# Add arguments for inspect_environment.py
209218
args.append(directory)
210-
219+
211220
if not require_requirements_txt:
212221
args.append("--no-require-requirements")
213222

rsconnect/subprocesses/inspect_environment.py

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,12 @@ class EnvironmentException(Exception):
6060
pass
6161

6262

63-
def detect_environment(dirname: str, force_generate: bool = False, require_requirements_txt: bool = True) -> EnvironmentData:
63+
def detect_environment(
64+
dirname: str, force_generate: bool = False, require_requirements_txt: bool = True
65+
) -> EnvironmentData:
6466
"""Determine the python dependencies in the environment.
6567
66-
`pip freeze` will be used to introspect the environment if force_generate=True or if
68+
`pip freeze` will be used to introspect the environment if force_generate=True or if
6769
requirements.txt is missing and require_requirements_txt=False.
6870
6971
:param: dirname Directory name
@@ -218,33 +220,34 @@ def main():
218220
try:
219221
if len(sys.argv) < 2:
220222
raise EnvironmentException("Usage: %s [-fc] DIRECTORY [--no-require-requirements]" % sys.argv[0])
221-
223+
222224
# Parse arguments
223225
flags = ""
224226
force_generate = False
225227
require_requirements_txt = True
226-
228+
227229
# Check for flags in first argument
228-
if len(sys.argv) > 2 and sys.argv[1].startswith('-') and not sys.argv[1].startswith('--'):
230+
if len(sys.argv) > 2 and sys.argv[1].startswith("-") and not sys.argv[1].startswith("--"):
229231
flags = sys.argv[1]
230232
if "f" in flags:
231233
force_generate = True
232-
234+
233235
# Check for --no-require-requirements flag
234236
if "--no-require-requirements" in sys.argv:
235237
require_requirements_txt = False
236-
238+
237239
# directory is always the first non-flag argument
238240
directory_index = 1
239-
while directory_index < len(sys.argv) and (sys.argv[directory_index].startswith('-') or
240-
sys.argv[directory_index] == "--no-require-requirements"):
241+
while directory_index < len(sys.argv) and (
242+
sys.argv[directory_index].startswith("-") or sys.argv[directory_index] == "--no-require-requirements"
243+
):
241244
directory_index += 1
242-
245+
243246
if directory_index >= len(sys.argv):
244247
raise EnvironmentException("Missing directory argument")
245-
248+
246249
directory = sys.argv[directory_index]
247-
250+
248251
envinfo = detect_environment(directory, force_generate, require_requirements_txt)._asdict()
249252
if "contents" in envinfo:
250253
keepers = list(map(strip_ref, envinfo["contents"].split("\n")))

tests/test_environment.py

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -83,13 +83,13 @@ def test_pip_freeze(self):
8383
python_interpreter=sys.executable,
8484
)
8585
self.assertEqual(expected, result)
86-
86+
8787
def test_missing_requirements_file(self):
8888
"""Test that missing requirements.txt raises an exception"""
8989
with tempfile.TemporaryDirectory() as empty_dir:
9090
with self.assertRaises(RSConnectException) as context:
9191
Environment.create_python_environment(empty_dir)
92-
92+
9393
self.assertIn("requirements.txt file is required", str(context.exception))
9494

9595
def test_filter_pip_freeze_output(self):
@@ -144,22 +144,30 @@ def test_is_not_executable(self):
144144

145145
class TestPythonVersionRequirements:
146146
def test_pyproject_toml(self):
147-
env = Environment.create_python_environment(os.path.join(TESTDATA, "python-project", "using_pyproject"), require_requirements_txt=False)
147+
env = Environment.create_python_environment(
148+
os.path.join(TESTDATA, "python-project", "using_pyproject"), require_requirements_txt=False
149+
)
148150
assert env.python_interpreter == sys.executable
149151
assert env.python_version_requirement == ">=3.8"
150152

151153
def test_python_version(self):
152-
env = Environment.create_python_environment(os.path.join(TESTDATA, "python-project", "using_pyversion"), require_requirements_txt=False)
154+
env = Environment.create_python_environment(
155+
os.path.join(TESTDATA, "python-project", "using_pyversion"), require_requirements_txt=False
156+
)
153157
assert env.python_interpreter == sys.executable
154158
assert env.python_version_requirement == ">=3.8,<3.12"
155159

156160
def test_all_of_them(self):
157-
env = Environment.create_python_environment(os.path.join(TESTDATA, "python-project", "allofthem"), require_requirements_txt=False)
161+
env = Environment.create_python_environment(
162+
os.path.join(TESTDATA, "python-project", "allofthem"), require_requirements_txt=False
163+
)
158164
assert env.python_interpreter == sys.executable
159165
assert env.python_version_requirement == ">=3.8,<3.12"
160166

161167
def test_missing(self):
162-
env = Environment.create_python_environment(os.path.join(TESTDATA, "python-project", "empty"), require_requirements_txt=False)
168+
env = Environment.create_python_environment(
169+
os.path.join(TESTDATA, "python-project", "empty"), require_requirements_txt=False
170+
)
163171
assert env.python_interpreter == sys.executable
164172
assert env.python_version_requirement is None
165173

@@ -281,6 +289,7 @@ def fake_inspect_environment(
281289
assert environment.python_interpreter == expected_python
282290
assert environment == expected_environment
283291

292+
284293
class TestEnvironmentDeprecations:
285294
def test_override_python_version(self):
286295
with mock.patch.object(rsconnect.environment.logger, "warning") as mock_warning:

tests/test_subprocesses_inspect_environment.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,10 @@
1-
import os
21
import tempfile
32
import pytest
43
from unittest import mock
54

6-
from rsconnect.exception import RSConnectException
75
from rsconnect.subprocesses.inspect_environment import (
86
output_file,
97
detect_environment,
10-
pip_freeze,
118
EnvironmentException,
129
)
1310

@@ -17,7 +14,7 @@ def test_output_file_requires_requirements_txt():
1714
with tempfile.TemporaryDirectory() as empty_dir:
1815
with pytest.raises(EnvironmentException) as context:
1916
output_file(empty_dir, "requirements.txt", "pip")
20-
17+
2118
assert "requirements.txt file is required" in str(context.value)
2219

2320

@@ -26,14 +23,14 @@ def test_detect_environment_requires_requirements_txt():
2623
with tempfile.TemporaryDirectory() as empty_dir:
2724
with pytest.raises(EnvironmentException) as context:
2825
detect_environment(empty_dir, force_generate=False)
29-
26+
3027
assert "requirements.txt file is required" in str(context.value)
3128

3229

3330
def test_detect_environment_with_force_generate():
3431
"""Test that detect_environment still works with force_generate=True"""
3532
with tempfile.TemporaryDirectory() as empty_dir:
36-
with mock.patch('rsconnect.subprocesses.inspect_environment.pip_freeze') as mock_pip_freeze:
33+
with mock.patch("rsconnect.subprocesses.inspect_environment.pip_freeze") as mock_pip_freeze:
3734
mock_pip_freeze.return_value = {
3835
"filename": "requirements.txt",
3936
"contents": "numpy\npandas",

0 commit comments

Comments
 (0)