API-Manager/apimanager/apicollectionlist/views.py

from django.shortcuts import render

# Create your views here.
# -*- coding: utf-8 -*-
"""
Views of Api Collection list app
"""
import datetime
from django.contrib import messages
from django.contrib.auth.mixins import LoginRequiredMixin
import json
from django.urls import reverse_lazy
from django.http import HttpResponse
from django.views.generic import FormView,TemplateView, View
from apicollections.views import IndexView
from obp.api import API, APIError
from django.conf import settings
import csv



class ApiCollectionListView(IndexView, LoginRequiredMixin, FormView ):
    template_name = "apicollectionlist/apicollectionlist.html"
    success_url = '/apicollections/list'

    def get_apicollections(self, context):
        """Get All user """
        api = API(self.request.session.get('obp'))
        try:
            apicollections_list = []
            #TODO: NOTE- THIS RETURNS ALL USER. THIS IS A POTENTIAL PERFORMANCE ISSUE. WE NEED ENDPOINT FOR COLLECTION.
            #Endpoint will return users.
            get_all_users_url_path = '/users'

            user = api.get(get_all_users_url_path)
            for i in user["users"]:
                # Returns the APIs collections for a user.
                api_collections_for_user_url_path = '/users/{}/api-collections'.format(i["user_id"])
                api_collections_for_user = api.get(api_collections_for_user_url_path)
                if 'api_collections' in api_collections_for_user:
                   apicollections_list.extend(api_collections_for_user['api_collections'])
            for ac in apicollections_list:
                ac["collection_on_api_explorer_url"] = f"{settings.API_EXPLORER_HOST}/?api-collection-id={ac['api_collection_id']}"
        except APIError as err:
            messages.error(self.request, err)
            return []
        except Exception as err:
            messages.error(self.request, err)
            return []

        return apicollections_list

    def get_context_data(self, **kwargs):
        api = API(self.request.session.get('obp'))
        context = super(IndexView, self).get_context_data(**kwargs)
        apicollections_list = self.get_apicollections(context)
        try:
            for final_collection_list in apicollections_list:
                url_path = "/users/user_id/{}".format(final_collection_list["user_id"])
                result =  api.get(url_path)
                final_collection_list["username"] = result["username"]
        except Exception as err:
            messages.error(self.request, err)
        context.update({
            'apicollections_list': apicollections_list,
        })
        return context

class ExportCsvView(LoginRequiredMixin, View):
    """View to export the user to csv"""

    def get(self, request, *args, **kwargs):
       api = API(self.request.session.get('obp'))
       try:
           apicollections_list = []
           urlpath = '/my/api-collections'
           result = api.get(urlpath)
           if 'api_collections' in result:
               apicollections_list.extend(result['api_collections'])
       except APIError as err:
           messages.error(self.request, err)
       except Exception as inst:
           messages.error(self.request, "Unknown Error {}".format(type(inst).__name__))
       response = HttpResponse(content_type = 'text/csv')
       response['Content-Disposition'] = 'attachment;filename= ApiCollections'+ str(datetime.datetime.now())+'.csv'
       writer = csv.writer(response)
       writer.writerow(["api_collection_id","user_id","api_collection_name","is_sharable","description"])
       for user in apicollections_list:
          writer.writerow([user['api_collection_id'],user['user_id'], user['api_collection_name'], user["is_sharable"],
                             user["description"]])
       return response