ap: more strict filename matching

This commit is contained in:
Vladislav Yarmak 2019-06-09 21:34:57 +03:00
parent 0d0f9c8bf2
commit cc67638603

View File

@ -54,6 +54,9 @@ class ExtractException(Exception):
class PatternNotFoundException(Exception): class PatternNotFoundException(Exception):
pass pass
class UnknownPlatformException(Exception):
pass
class ExtractedTarget: class ExtractedTarget:
name = None name = None
@ -173,8 +176,11 @@ def main():
os_prefix = 'ws2012_x64' os_prefix = 'ws2012_x64'
elif 'winserv-2016' in installer_name: elif 'winserv-2016' in installer_name:
os_prefix = 'ws2016_x64' os_prefix = 'ws2016_x64'
else: elif 'win10' in installer_name:
os_prefix = 'win10_x64' os_prefix = 'win10_x64'
else:
raise UnknownPlatformException("Can't infer platform from filename %s"
% (repr(installer_name),))
driver_name = drv_prefix[product_type] + version driver_name = drv_prefix[product_type] + version
out_dir = os.path.join( out_dir = os.path.join(
os.path.dirname( os.path.dirname(