Skip to content

Commit 12182ff

Browse files
committed
Rename prefixes model to Prefix
Changes to be committed: modified: api/admin.py modified: api/groups.py new file: api/migrations/0005_rename_prefixes_prefix.py modified: api/models.py modified: api/scripts/method_specific/POST_api_prefixes_create.py modified: api/scripts/method_specific/POST_api_prefixes_delete.py modified: api/scripts/method_specific/POST_api_prefixes_modify.py modified: api/scripts/method_specific/POST_api_prefixes_permissions_set.py modified: api/scripts/utilities/UserUtils.py
1 parent 0252a50 commit 12182ff

File tree

9 files changed

+54
-62
lines changed

9 files changed

+54
-62
lines changed

bco_api/api/admin.py

Lines changed: 10 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1,29 +1,13 @@
1-
from django.contrib import admin
2-
3-
# Object-level permissions.
4-
# Source: https://github.com/django-guardian/django-guardian#admin-integration
5-
from guardian.admin import GuardedModelAdmin
6-
7-
from .models import bco, prefix_table, new_users, prefixes
1+
"""Django Admin
82
9-
class BcoDraftAdmin(
10-
GuardedModelAdmin
11-
):
12-
pass
3+
Registers models for the Django Admin app
4+
"""
135

14-
admin.site.register(
15-
bco,
16-
BcoDraftAdmin
17-
)
18-
19-
admin.site.register(
20-
prefix_table
21-
)
22-
23-
admin.site.register(
24-
new_users
25-
)
6+
from django.contrib import admin
7+
from guardian.admin import GuardedModelAdmin
8+
from api.models import bco, prefix_table, new_users, Prefix
269

27-
admin.site.register(
28-
prefixes
29-
)
10+
admin.site.register(bco, GuardedModelAdmin)
11+
admin.site.register(prefix_table)
12+
admin.site.register(new_users)
13+
admin.site.register(Prefix)

bco_api/api/groups.py

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
#!/usr/bin/env python3
22
"""Functions for operations with groups
33
"""
4+
45
from django.db import models
56
import django.db.utils as PermErrors
67
from django.db.models.signals import post_save, post_delete, pre_delete
@@ -11,7 +12,6 @@
1112
from django.utils import timezone
1213
from guardian.shortcuts import assign_perm
1314
from rest_framework import status
14-
from rest_framework.authtoken.models import Token
1515
from rest_framework.response import Response
1616

1717
from api.scripts.utilities import DbUtils
@@ -26,23 +26,11 @@ class GroupInfo(models.Model):
2626
"""
2727

2828
delete_members_on_group_deletion = models.BooleanField(default=False)
29-
30-
# The group
3129
group = models.OneToOneField(Group, on_delete=models.CASCADE)
32-
# Delete group members on group deletion?
33-
delete_members_on_group_deletion = models.BooleanField(default=False)
34-
# Description of the group
3530
description = models.TextField(blank = True)
36-
# Expiration date
3731
expiration = models.DateTimeField(blank=True, null=True)
38-
39-
# The group
4032
group = models.ForeignKey(Group, on_delete=models.CASCADE, to_field='name')
41-
42-
# Size limit for the number of members
4333
max_n_members = models.IntegerField(blank=True, null=True)
44-
45-
# Which user owns it?
4634
owner_user = models.ForeignKey(User, on_delete=models.CASCADE, to_field='username')
4735

4836
# --- Group --- #
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
# Generated by Django 3.2.10 on 2022-03-22 18:29
2+
3+
from django.conf import settings
4+
from django.db import migrations
5+
6+
7+
class Migration(migrations.Migration):
8+
9+
dependencies = [
10+
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
11+
('auth', '0012_alter_user_first_name_max_length'),
12+
('api', '0004_rename_group_info_groupinfo'),
13+
]
14+
15+
operations = [
16+
migrations.RenameModel(
17+
old_name='prefixes',
18+
new_name='Prefix',
19+
),
20+
]

bco_api/api/models.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -114,13 +114,13 @@ class new_users(models.Model):
114114
created = models.DateTimeField(default=timezone.now)
115115

116116

117-
# Link prefixes to groups and users.
117+
# Link Prefix to groups and users.
118118

119119
# Be careful about related_name.
120120
# Source: https://stackoverflow.com/questions/53651114/using-same-foreign-key-twice-in-a-model-in-django-as-different-fields
121-
class prefixes(models.Model):
122-
123-
# Which server is this prefix certified with?
121+
class Prefix(models.Model):
122+
"""Which server is this prefix certified with?"""
123+
124124
certifying_server = models.TextField(blank = True, null = True)
125125

126126
# What is the certifying key?
@@ -162,7 +162,7 @@ class prefixes(models.Model):
162162

163163
def __str__(self):
164164
"""String for representing the BCO model (in Admin site etc.)."""
165-
return self.prefix
165+
return "{}".format(self.prefix)
166166

167167
# def get_first_name(self):
168168
# return self.first_name
@@ -224,7 +224,7 @@ def create_auth_token(sender, instance=None, created=False, **kwargs):
224224
Token.objects.create(user=instance)
225225

226226
# --- Prefix --- #
227-
@receiver(post_save, sender=prefixes)
227+
@receiver(post_save, sender=Prefix)
228228
def create_permissions_for_prefix(sender, instance=None, created=False, **kwargs):
229229
"""Link prefix creation to permissions creation.
230230
Check to see whether or not the permissions
@@ -286,7 +286,7 @@ def create_permissions_for_prefix(sender, instance=None, created=False, **kwargs
286286
pass
287287

288288

289-
@receiver(post_delete, sender=prefixes)
289+
@receiver(post_delete, sender=Prefix)
290290
def delete_permissions_for_prefix(sender, instance=None, **kwargs):
291291
"""Link prefix deletion to permissions deletion.
292292
No risk of raising an error when using

bco_api/api/scripts/method_specific/POST_api_prefixes_create.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
from api.scripts.utilities import DbUtils
1212
from api.scripts.utilities import UserUtils
13-
from api.models import prefixes
13+
from api.models import Prefix
1414

1515
from rest_framework import status
1616
from rest_framework.response import Response
@@ -46,7 +46,7 @@ def POST_api_prefixes_create(request):
4646

4747
# Get all existing prefixes.
4848
available_prefixes = list(
49-
prefixes.objects.all().values_list('prefix', flat = True))
49+
Prefix.objects.all().values_list('prefix', flat = True))
5050

5151
# Construct an array to return information about processing
5252
# the request.
@@ -118,7 +118,7 @@ def POST_api_prefixes_create(request):
118118
# The prefix has not been created, so create it.
119119
DbUtils.DbUtils().write_object(
120120
p_app_label = 'api',
121-
p_model_name = 'prefixes',
121+
p_model_name = 'Prefix',
122122
p_fields = ['created_by', 'description', 'owner_group', 'owner_user', 'prefix'],
123123
p_data = {
124124
'created_by': user_utils.user_from_request(

bco_api/api/scripts/method_specific/POST_api_prefixes_delete.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
objects.
99
"""
1010
from api.scripts.utilities import DbUtils
11-
from api.models import prefixes
11+
from api.models import Prefix
1212
from rest_framework import status
1313
from rest_framework.response import Response
1414

@@ -33,7 +33,7 @@ def POST_api_prefixes_delete(request):
3333

3434
# Get all existing prefixes.
3535
available_prefixes = list(
36-
prefixes.objects.all().values_list('prefix', flat=True))
36+
Prefix.objects.all().values_list('prefix', flat=True))
3737

3838
returning = []
3939

@@ -60,7 +60,7 @@ def POST_api_prefixes_delete(request):
6060
# just delete straight up.
6161
# Source: https://stackoverflow.com/a/3681691
6262
# Django *DOESN'T* want primary keys now...
63-
prefixed = prefixes.objects.get(prefix=standardized)
63+
prefixed = Prefix.objects.get(prefix=standardized)
6464
prefixed.delete()
6565
# Deleted the prefix.
6666
errors['200_OK_prefix_delete'] = db_utils.messages(parameters={

bco_api/api/scripts/method_specific/POST_api_prefixes_modify.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
"""
88
from api.scripts.utilities import DbUtils
99
from api.scripts.utilities import UserUtils
10-
from api.models import prefixes
10+
from api.models import Prefix
1111
from rest_framework import status
1212
from rest_framework.response import Response
1313

@@ -34,7 +34,7 @@ def POST_api_prefixes_modify(request):
3434

3535
bulk_request = request.data['POST_api_prefixes_modify']
3636
available_prefixes = list(
37-
prefixes.objects.all().values_list('prefix', flat = True))
37+
Prefix.objects.all().values_list('prefix', flat = True))
3838

3939
# Construct an array to return information about processing
4040
# the request.
@@ -112,7 +112,7 @@ def POST_api_prefixes_modify(request):
112112
# The prefix has not been created, so create it.
113113
DbUtils.DbUtils().write_object(
114114
p_app_label = 'api',
115-
p_model_name = 'prefixes',
115+
p_model_name = 'Prefix',
116116
p_fields = ['created_by', 'description', 'owner_group', 'owner_user', 'prefix'],
117117
p_data = {
118118
'created_by': user_utils.user_from_request(

bco_api/api/scripts/method_specific/POST_api_prefixes_permissions_set.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# Prefixes
2-
from ...models import prefixes
2+
from api.models import Prefix
33

44
# DB utilities
55
from ..utilities import DbUtils
@@ -37,7 +37,7 @@ def POST_api_prefixes_permissions_set(
3737

3838
# Get all existing prefixes.
3939
available_prefixes = list(
40-
prefixes.objects.all().values_list(
40+
Prefix.objects.all().values_list(
4141
'prefix',
4242
flat = True
4343
)

bco_api/api/scripts/utilities/UserUtils.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
# Prefixes
2-
from ...models import prefixes
1+
# Prefix
2+
from api.models import Prefix
33

44
# For returning server information.
55
from django.conf import settings
@@ -102,7 +102,7 @@ def check_user_owns_prefix(
102102
):
103103

104104
# Check if a user owns a prefix.
105-
return prefixes.objects.filter(owner_user=un, prefix=prfx).exists()
105+
return Prefix.objects.filter(owner_user=un, prefix=prfx).exists()
106106

107107
def get_user_groups_by_token(
108108
self,
@@ -253,7 +253,7 @@ def get_user_info(
253253
'other_info' : other_info
254254
}
255255

256-
# Prefixes for a given user.
256+
# Prefix for a given user.
257257
def prefixes_for_user(
258258
self,
259259
user_object
@@ -285,7 +285,7 @@ def prefix_perms_for_user(self, user_object, flatten=True, specific_permission=N
285285

286286
# Get the prefixes for the user and their groups, then filter.
287287
# pxs = list(
288-
# prefixes.objects.filter(
288+
# Prefix.objects.filter(
289289
# Q(owner_user = user_object.id) |
290290
# Q(owner_group__in = list(user_object.groups.all().values_list(
291291
# 'id',

0 commit comments

Comments
 (0)