mirror of
https://github.com/actions/setup-python.git
synced 2024-11-06 07:55:52 -05:00
22 lines
1,000 B
Python
22 lines
1,000 B
Python
|
import sys
|
||
|
|
argCount = len(sys.argv) - 1
|
||
|
|
|
||
|
|
if argCount == 1:
|
||
|
|
expectedVersion = sys.argv[1]
|
||
|
|
versions = len(expectedVersion.split("."))
|
||
|
|
majorMinor = str(sys.version_info[0]) + '.' + str(sys.version_info[1])
|
||
|
|
|
||
|
|
if versions == 2:
|
||
|
|
# Test only major and minor version
|
||
|
|
if expectedVersion != majorMinor:
|
||
|
|
raise Exception("Incorrect major + minor version detected\nExpected: " + expectedVersion + "\nActual: " + majorMinor)
|
||
|
|
elif versions == 3:
|
||
|
|
# Test major, minor and micro version
|
||
|
|
majorMinorMicro = majorMinor + '.' + str(sys.version_info[2])
|
||
|
|
if expectedVersion != majorMinorMicro:
|
||
|
|
raise Exception("Incorrect major + minor + micro version detected\nExpected: " + expectedVersion + "\nActual: " + majorMinorMicro)
|
||
|
|
else:
|
||
|
|
raise Exception("Incorrect number of arguments supplied")
|
||
|
|
print("Correct version of Python " + expectedVersion + " detected")
|
||
|
|
else:
|
||
|
|
raise Exception("Incorrect number of arguments supplied")
|