greenbone /
python-gvm
| @@ 420-451 (lines=32) @@ | ||
| 417 | if comment: |
|
| 418 | cmd.add_element("comment", comment) |
|
| 419 | ||
| 420 | if resource_id or resource_type: |
|
| 421 | if not resource_id: |
|
| 422 | raise RequiredArgument( |
|
| 423 | function=self.create_permission.__name__, |
|
| 424 | argument='resource_id', |
|
| 425 | ) |
|
| 426 | ||
| 427 | if not resource_type: |
|
| 428 | raise RequiredArgument( |
|
| 429 | function=self.create_permission.__name__, |
|
| 430 | argument='resource_type', |
|
| 431 | ) |
|
| 432 | ||
| 433 | if not isinstance(resource_type, self.types.EntityType): |
|
| 434 | raise InvalidArgumentType( |
|
| 435 | function=self.create_permission.__name__, |
|
| 436 | argument='resource_type', |
|
| 437 | arg_type=self.types.EntityType.__name__, |
|
| 438 | ) |
|
| 439 | ||
| 440 | _xmlresource = cmd.add_element( |
|
| 441 | "resource", attrs={"id": resource_id} |
|
| 442 | ) |
|
| 443 | ||
| 444 | _actual_resource_type = resource_type |
|
| 445 | if resource_type.value == EntityType.AUDIT.value: |
|
| 446 | _actual_resource_type = EntityType.TASK |
|
| 447 | elif resource_type.value == EntityType.POLICY.value: |
|
| 448 | _actual_resource_type = EntityType.SCAN_CONFIG |
|
| 449 | ||
| 450 | _xmlresource.add_element("type", _actual_resource_type.value) |
|
| 451 | ||
| 452 | return self._send_xml_command(cmd) |
|
| 453 | ||
| 454 | def create_policy( |
|
| @@ 925-953 (lines=29) @@ | ||
| 922 | if name: |
|
| 923 | cmd.add_element("name", name) |
|
| 924 | ||
| 925 | if resource_id or resource_type: |
|
| 926 | if not resource_id: |
|
| 927 | raise RequiredArgument( |
|
| 928 | function=self.modify_permission.__name__, |
|
| 929 | argument='resource_id', |
|
| 930 | ) |
|
| 931 | ||
| 932 | if not resource_type: |
|
| 933 | raise RequiredArgument( |
|
| 934 | function=self.modify_permission.__name__, |
|
| 935 | argument='resource_type', |
|
| 936 | ) |
|
| 937 | ||
| 938 | if not isinstance(resource_type, self.types.EntityType): |
|
| 939 | raise InvalidArgumentType( |
|
| 940 | function=self.modify_permission.__name__, |
|
| 941 | argument='resource_type', |
|
| 942 | arg_type=self.types.EntityType.__name__, |
|
| 943 | ) |
|
| 944 | ||
| 945 | _xmlresource = cmd.add_element( |
|
| 946 | "resource", attrs={"id": resource_id} |
|
| 947 | ) |
|
| 948 | _actual_resource_type = resource_type |
|
| 949 | if resource_type.value == EntityType.AUDIT.value: |
|
| 950 | _actual_resource_type = EntityType.TASK |
|
| 951 | elif resource_type.value == EntityType.POLICY.value: |
|
| 952 | _actual_resource_type = EntityType.SCAN_CONFIG |
|
| 953 | _xmlresource.add_element("type", _actual_resource_type.value) |
|
| 954 | ||
| 955 | if subject_id or subject_type: |
|
| 956 | if not subject_id: |
|
