from PLC.Method import Method
from PLC.Parameter import Parameter, Mixed
from PLC.Filter import Filter
from PLC.Auth import Auth
from PLC.Slices import Slice, Slices
from PLC.Sites import Site, Sites

class SliceExtendedInfo(Method):
    """
    Deprecated. Can be implemented with GetSlices.

    Returns an array of structs containing details about slices. 
    The summary can optionally include the list of nodes in and 
    users of each slice.

    Users may only query slices of which they are members. PIs may
    query any of the slices at their sites. Admins may query any
    slice. If a slice that cannot be queried is specified in
    slice_filter, details about that slice will not be returned.
    """

    roles = ['admin', 'pi', 'user']

    accepts = [
        Auth(),
        [Slice.fields['name']],
        Parameter(bool, "Whether or not to return users for the slices", nullok = True),
	Parameter(bool, "Whether or not to return nodes for the slices", nullok = True)
        ]

    returns = [Slice.fields]
    

    def call(self, auth, slice_name_list=None, return_users=None, return_nodes=None):
	# If we are not admin, make sure to return only viewable
	# slices.
	slice_filter = slice_name_list
	slices = Slices(self.api, slice_filter)
	if not slices:
            raise PLCInvalidArgument, "No such slice"

        if 'admin' not in self.caller['roles']:
            # Get slices that we are able to view
            valid_slice_ids = self.caller['slice_ids']
            if 'pi' in self.caller['roles'] and self.caller['site_ids']:
                sites = Sites(self.api, self.caller['site_ids'])
                for site in sites:
                    valid_slice_ids += site['slice_ids']

            if not valid_slice_ids:
                return []
	  
	    slices = filter(lambda slice: slice['slice_id'] in valid_slice_ids, slices)

	for slice in slices:
	    slices.pop(slice)
	    person_ids = slice.pop('person_ids')
	    node_ids = slice.pop('node_ids')
	    if return_users:
		slice['users'] = person_ids
	    if return_nodes:
	        slice['nodes'] = node_ids
	    slices.add(slice)
		
	
        return slices