wdev.py: added support for finding all available VS installations.

Author: julian-smith-artifex-com

wdev.py

@@ -13,6 +13,7 @@
 
 import pipcl
 
+
 class WindowsVS:
     r'''
     Windows only. Finds locations of Visual Studio command-line tools. Assumes
@@ -33,7 +34,16 @@ class WindowsVS:
 
     `.csc` is C# compiler; will be None if not found.
     '''
-    def __init__( self, year=None, grade=None, version=None, cpu=None, verbose=False):
+    def __init__(
+            self,
+            *,
+            year=None,
+            grade=None,
+            version=None,
+            cpu=None,
+            directory=None,
+            verbose=False,
+            ):
         '''
         Args:
             year:
@@ -52,7 +62,15 @@ def __init__( self, year=None, grade=None, version=None, cpu=None, verbose=False
                 variable WDEV_VS_VERSION if set.
             cpu:
                 None or a `WindowsCpu` instance.
+            directory:
+                Ignore year, grade, version and cpu and use this directory
+                directly.
+            verbose:
+                .
+            
         '''
+        if year is not None:
+            year = str(year)    # Allow specification as a number.
         def default(value, name):
             if value is None:
                 name2 = f'WDEV_VS_{name.upper()}'
@@ -68,16 +86,17 @@ def default(value, name):
             if not cpu:
                 cpu = WindowsCpu()
 
-            # Find `directory`.
-            #
-            pattern = f'C:\\Program Files*\\Microsoft Visual Studio\\{year if year else "2*"}\\{grade if grade else "*"}'
-            directories = glob.glob( pattern)
-            if verbose:
-                _log( f'Matches for: {pattern=}')
-                _log( f'{directories=}')
-            assert directories, f'No match found for: {pattern}'
-            directories.sort()
-            directory = directories[-1]
+            if not directory:
+                # Find `directory`.
+                #
+                pattern = _vs_pattern(year, grade)
+                directories = glob.glob( pattern)
+                if verbose:
+                    _log( f'Matches for: {pattern=}')
+                    _log( f'{directories=}')
+                assert directories, f'No match found for {pattern=}.'
+                directories.sort()
+                directory = directories[-1]
 
             # Find `devenv`.
             #
@@ -167,7 +186,7 @@ def default(value, name):
             self.year = year
             self.cpu = cpu
         except Exception as e:
-            raise Exception( f'Unable to find Visual Studio') from e
+            raise Exception( f'Unable to find Visual Studio {year=} {grade=} {version=} {cpu=} {directory=}') from e
 
     def description_ml( self, indent=''):
         '''
@@ -189,7 +208,40 @@ def description_ml( self, indent=''):
         return textwrap.indent( ret, indent)
 
     def __repr__( self):
-        return ' '.join( self._description())
+        items = list()
+        for name in (
+                'year',
+                'grade',
+                'version',
+                'directory',
+                'vcvars',
+                'cl',
+                'link',
+                'csc',
+                'msbuild',
+                'devenv',
+                'cpu',
+                ):
+            items.append(f'{name}={getattr(self, name)!r}')
+        return ' '.join(items)
+
+
+def _vs_pattern(year=None, grade=None):
+    return f'C:\\Program Files*\\Microsoft Visual Studio\\{year if year else "2*"}\\{grade if grade else "*"}'
+
+
+def windows_vs_multiple(year=None, grade=None, verbose=0):
+    '''
+    Returns list of WindowsVS instances.
+    '''
+    ret = list()
+    directories = glob.glob(_vs_pattern(year, grade))
+    for directory in directories:
+        vs = WindowsVS(directory=directory)
+        if verbose:
+            _log(vs.description_ml())
+        ret.append(vs)
+    return ret
 
 
 class WindowsCpu: