greenbone /
python-gvm
| @@ 131-161 (lines=31) @@ | ||
| 128 | if comment: |
|
| 129 | cmd.add_element("comment", comment) |
|
| 130 | ||
| 131 | if resource_id or resource_type: |
|
| 132 | if not resource_id: |
|
| 133 | raise RequiredArgument( |
|
| 134 | function=self.create_permission.__name__, |
|
| 135 | argument='resource_id', |
|
| 136 | ) |
|
| 137 | ||
| 138 | if not resource_type: |
|
| 139 | raise RequiredArgument( |
|
| 140 | function=self.create_permission.__name__, |
|
| 141 | argument='resource_type', |
|
| 142 | ) |
|
| 143 | ||
| 144 | if not isinstance(resource_type, EntityType): |
|
| 145 | raise InvalidArgumentType( |
|
| 146 | function=self.create_permission.__name__, |
|
| 147 | argument='resource_type', |
|
| 148 | arg_type=EntityType.__name__, |
|
| 149 | ) |
|
| 150 | ||
| 151 | _xmlresource = cmd.add_element( |
|
| 152 | "resource", attrs={"id": resource_id} |
|
| 153 | ) |
|
| 154 | ||
| 155 | _actual_resource_type = resource_type |
|
| 156 | if resource_type.value == EntityType.AUDIT.value: |
|
| 157 | _actual_resource_type = EntityType.TASK |
|
| 158 | elif resource_type.value == EntityType.POLICY.value: |
|
| 159 | _actual_resource_type = EntityType.SCAN_CONFIG |
|
| 160 | ||
| 161 | _xmlresource.add_element("type", _actual_resource_type.value) |
|
| 162 | ||
| 163 | return self._send_xml_command(cmd) |
|
| 164 | ||
| @@ 272-300 (lines=29) @@ | ||
| 269 | if name: |
|
| 270 | cmd.add_element("name", name) |
|
| 271 | ||
| 272 | if resource_id or resource_type: |
|
| 273 | if not resource_id: |
|
| 274 | raise RequiredArgument( |
|
| 275 | function=self.modify_permission.__name__, |
|
| 276 | argument='resource_id', |
|
| 277 | ) |
|
| 278 | ||
| 279 | if not resource_type: |
|
| 280 | raise RequiredArgument( |
|
| 281 | function=self.modify_permission.__name__, |
|
| 282 | argument='resource_type', |
|
| 283 | ) |
|
| 284 | ||
| 285 | if not isinstance(resource_type, EntityType): |
|
| 286 | raise InvalidArgumentType( |
|
| 287 | function=self.modify_permission.__name__, |
|
| 288 | argument='resource_type', |
|
| 289 | arg_type=EntityType.__name__, |
|
| 290 | ) |
|
| 291 | ||
| 292 | _xmlresource = cmd.add_element( |
|
| 293 | "resource", attrs={"id": resource_id} |
|
| 294 | ) |
|
| 295 | _actual_resource_type = resource_type |
|
| 296 | if resource_type.value == EntityType.AUDIT.value: |
|
| 297 | _actual_resource_type = EntityType.TASK |
|
| 298 | elif resource_type.value == EntityType.POLICY.value: |
|
| 299 | _actual_resource_type = EntityType.SCAN_CONFIG |
|
| 300 | _xmlresource.add_element("type", _actual_resource_type.value) |
|
| 301 | ||
| 302 | if subject_id or subject_type: |
|
| 303 | if not subject_id: |
|
