From 4cc9426d72fe9baa8a4a7f56c6d43493bd011f02 Mon Sep 17 00:00:00 2001 From: Trevor Bramwell Date: Wed, 15 Jul 2020 13:00:53 -0700 Subject: [PATCH] Fix Flake8 Violations Several minor issues found through flake8 should now be fixed in the code. Signed-off-by: Trevor Bramwell Change-Id: I4f6e5dac5ed8378911598ded714602bbb3f6bd6b --- src/account/models.py | 3 ++- src/account/tasks.py | 2 +- src/account/views.py | 2 +- src/api/migrations/0013_manual_20200218_1536.py | 2 +- src/api/models.py | 12 ++++++------ src/api/views.py | 2 ++ src/resource_inventory/migrations/0012_auto_20200103_1850.py | 2 +- src/workflow/resource_bundle_workflow.py | 4 +--- 8 files changed, 15 insertions(+), 14 deletions(-) diff --git a/src/account/models.py b/src/account/models.py index 4aab306..9624a53 100644 --- a/src/account/models.py +++ b/src/account/models.py @@ -16,6 +16,7 @@ import random from collections import Counter + class LabStatus(object): """ A Poor man's enum for the status of a lab. @@ -216,7 +217,7 @@ class Lab(models.Model): def get_available_resources(self): # Cannot import model normally due to ciruclar import - Server = apps.get_model('resource_inventory', 'Server') # TODO: Find way to import ResourceQuery + Server = apps.get_model('resource_inventory', 'Server') # TODO: Find way to import ResourceQuery resources = [str(resource.profile) for resource in Server.objects.filter(lab=self, booked=False)] return dict(Counter(resources)) diff --git a/src/account/tasks.py b/src/account/tasks.py index 53fbaf5..df98c73 100644 --- a/src/account/tasks.py +++ b/src/account/tasks.py @@ -28,7 +28,7 @@ def sync_jira_accounts(): continue try: user.email = user_dict['emailAddress'] - except: + except KeyError: pass user.userprofile.url = user_dict['self'] user.userprofile.full_name = user_dict['displayName'] diff --git a/src/account/views.py b/src/account/views.py index 1cb2275..d11a199 100644 --- a/src/account/views.py +++ b/src/account/views.py @@ -130,7 +130,7 @@ class JiraAuthenticatedView(RedirectView): email = "" try: email = jira.user(username).emailAddress - except: + except AttributeError: email = "" url = '/' # Step 3. Lookup the user or create them if they don't exist. diff --git a/src/api/migrations/0013_manual_20200218_1536.py b/src/api/migrations/0013_manual_20200218_1536.py index 0b76e84..3ac427e 100644 --- a/src/api/migrations/0013_manual_20200218_1536.py +++ b/src/api/migrations/0013_manual_20200218_1536.py @@ -1,6 +1,6 @@ # Generated by Django 2.2 on 2020-02-18 15:36 -from django.db import migrations, models +from django.db import migrations class Migration(migrations.Migration): diff --git a/src/api/models.py b/src/api/models.py index addc02d..231ae31 100644 --- a/src/api/models.py +++ b/src/api/models.py @@ -176,12 +176,12 @@ class LabManager(object): def format_user(self, userprofile): return { - "id": userprofile.user.id, - "username": userprofile.user.username, - "email": userprofile.email_addr, - "first_name": userprofile.user.first_name, - "last_name": userprofile.user.last_name, - "company": userprofile.company + "id": userprofile.user.id, + "username": userprofile.user.username, + "email": userprofile.email_addr, + "first_name": userprofile.user.first_name, + "last_name": userprofile.user.last_name, + "company": userprofile.company } def get_users(self): diff --git a/src/api/views.py b/src/api/views.py index 4e0d058..75a0db3 100644 --- a/src/api/views.py +++ b/src/api/views.py @@ -97,11 +97,13 @@ def lab_status(request, lab_name=""): return JsonResponse(lab_manager.set_status(request.POST), safe=False) return JsonResponse(lab_manager.get_status(), safe=False) + def lab_users(request, lab_name=""): lab_token = request.META.get('HTTP_AUTH_TOKEN') lab_manager = LabManagerTracker.get(lab_name, lab_token) return HttpResponse(lab_manager.get_users(), content_type="text/plain") + def lab_user(request, lab_name="", user_id=-1): lab_token = request.META.get('HTTP_AUTH_TOKEN') lab_manager = LabManagerTracker.get(lab_name, lab_token) diff --git a/src/resource_inventory/migrations/0012_auto_20200103_1850.py b/src/resource_inventory/migrations/0012_auto_20200103_1850.py index 569e433..65d8f85 100644 --- a/src/resource_inventory/migrations/0012_auto_20200103_1850.py +++ b/src/resource_inventory/migrations/0012_auto_20200103_1850.py @@ -54,6 +54,6 @@ class Migration(migrations.Migration): model_name='vlan', name='network', field=models.ForeignKey(on_delete=django.db.models.deletion.DO_NOTHING, - to='resource_inventory.PhysicalNetwork', null=True), + to='resource_inventory.PhysicalNetwork', null=True), ), ] diff --git a/src/workflow/resource_bundle_workflow.py b/src/workflow/resource_bundle_workflow.py index 391d33e..fa05b22 100644 --- a/src/workflow/resource_bundle_workflow.py +++ b/src/workflow/resource_bundle_workflow.py @@ -10,6 +10,7 @@ from django.conf import settings from django.forms import formset_factory +from django.core.exceptions import ValidationError from typing import List @@ -251,15 +252,12 @@ class Define_Software(WorkflowStep): # TODO: fix headnode in form, currently doesn't return a selected one # models['headnode_index'] = post_data.get("headnode", 1) formset = self.create_hostformset(hosts, data=post_data) - has_headnode = False if formset.is_valid(): for i, form in enumerate(formset): host = hosts[i] image = form.cleaned_data['image'] hostname = form.cleaned_data['host_name'] headnode = form.cleaned_data['headnode'] - if headnode: - has_headnode = True host.is_head_node = headnode host.name = hostname host.image = image -- 2.16.6