<reponame>MosheSheena/ovirtlib4<gh_stars>0
# -*- coding: utf-8 -*-

import ovirtsdk4.types as types

from .system_service import CollectionService, CollectionEntity


class Disks(CollectionService):
    """
    Gives access to all Ovirt Disks
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.service = self.connection.system_service().disks_service()
        self.entity_service = self.service.disk_service
        self.entity_type = types.Disk

    def _get_collection_entity(self):
        """ Overwrite abstract parent method """
        return DiskEntity(connection=self.connection)


class DiskEntity(CollectionEntity):
    """
    Put Disk custom functions here
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

# -*- coding: utf-8 -*-

import ovirtsdk4.types as types

from .system_service import CollectionService, CollectionEntity


class Disks(CollectionService):
    """
    Gives access to all Ovirt Disks
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.service = self.connection.system_service().disks_service()
        self.entity_service = self.service.disk_service
        self.entity_type = types.Disk

    def _get_collection_entity(self):
        """ Overwrite abstract parent method """
        return DiskEntity(connection=self.connection)


class DiskEntity(CollectionEntity):
    """
    Put Disk custom functions here
    """
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
# -*- coding: utf-8 -*- Gives access to all Ovirt Disks Overwrite abstract parent method Put Disk custom functions here