Code Duplication    Length = 45-45 lines in 2 locations

gvm/protocols/gmpv208/types.py 1 location

@@ 479-523 (lines=45) @@
476
    GREENBONE_SENSOR_SCANNER_TYPE = "5"
477
478
479
def get_scanner_type_from_string(
480
    scanner_type: Optional[str],
481
) -> Optional[ScannerType]:
482
    """Convert a scanner type string to an actual ScannerType instance
483
484
    Arguments:
485
        scanner_type: Scanner type string to convert to a ScannerType
486
    """
487
    if not scanner_type:
488
        return None
489
490
    scanner_type = scanner_type.lower()
491
492
    if (
493
        scanner_type == ScannerType.OSP_SCANNER_TYPE.value
494
        or scanner_type == 'osp'
495
    ):
496
        return ScannerType.OSP_SCANNER_TYPE
497
498
    if (
499
        scanner_type == ScannerType.OPENVAS_SCANNER_TYPE.value
500
        or scanner_type == 'openvas'
501
    ):
502
        return ScannerType.OPENVAS_SCANNER_TYPE
503
504
    if (
505
        scanner_type == ScannerType.CVE_SCANNER_TYPE.value
506
        or scanner_type == 'cve'
507
    ):
508
        return ScannerType.CVE_SCANNER_TYPE
509
510
    if (
511
        scanner_type == ScannerType.GMP_SCANNER_TYPE.value
512
        or scanner_type == 'gmp'
513
    ):
514
        return ScannerType.GMP_SCANNER_TYPE
515
516
    if (
517
        scanner_type == ScannerType.GREENBONE_SENSOR_SCANNER_TYPE.value
518
        or scanner_type == 'greenbone'
519
    ):
520
        return ScannerType.GREENBONE_SENSOR_SCANNER_TYPE
521
522
    raise InvalidArgument(
523
        argument='scanner_type', function=get_scanner_type_from_string.__name__
524
    )
525
526

gvm/protocols/gmpv9/types.py 1 location

@@ 437-481 (lines=45) @@
434
    GREENBONE_SENSOR_SCANNER_TYPE = "5"
435
436
437
def get_scanner_type_from_string(
438
    scanner_type: Optional[str],
439
) -> Optional[ScannerType]:
440
    """Convert a scanner type string to an actual ScannerType instance
441
442
    Arguments:
443
        scanner_type: Scanner type string to convert to a ScannerType
444
    """
445
    if not scanner_type:
446
        return None
447
448
    scanner_type = scanner_type.lower()
449
450
    if (
451
        scanner_type == ScannerType.OSP_SCANNER_TYPE.value
452
        or scanner_type == 'osp'
453
    ):
454
        return ScannerType.OSP_SCANNER_TYPE
455
456
    if (
457
        scanner_type == ScannerType.OPENVAS_SCANNER_TYPE.value
458
        or scanner_type == 'openvas'
459
    ):
460
        return ScannerType.OPENVAS_SCANNER_TYPE
461
462
    if (
463
        scanner_type == ScannerType.CVE_SCANNER_TYPE.value
464
        or scanner_type == 'cve'
465
    ):
466
        return ScannerType.CVE_SCANNER_TYPE
467
468
    if (
469
        scanner_type == ScannerType.GMP_SCANNER_TYPE.value
470
        or scanner_type == 'gmp'
471
    ):
472
        return ScannerType.GMP_SCANNER_TYPE
473
474
    if (
475
        scanner_type == ScannerType.GREENBONE_SENSOR_SCANNER_TYPE.value
476
        or scanner_type == 'greenbone'
477
    ):
478
        return ScannerType.GREENBONE_SENSOR_SCANNER_TYPE
479
480
    raise InvalidArgument(
481
        argument='scanner_type', function=get_scanner_type_from_string.__name__
482
    )
483
484