From 83e279d1129fd7009d9722c010e5021d1788fcf0 Mon Sep 17 00:00:00 2001
From: ylsong <songyl157@chinaunicom.cn>
Date: Fri, 29 Dec 2023 09:43:36 +0800
Subject: [PATCH] add user and group group api

---
 backend/models.py                             |   7 ++++++
 backend/serializers.py                        |  14 +++++++++++
 backend/views.py                              |  22 ++++++++++++++++-
 .../__pycache__/__init__.cpython-39.pyc       | Bin 132 -> 136 bytes
 .../__pycache__/settings.cpython-39.pyc       | Bin 2354 -> 2655 bytes
 safeguard_web/__pycache__/urls.cpython-39.pyc | Bin 929 -> 845 bytes
 safeguard_web/settings.py                     |  23 ++++++++++++++++++
 safeguard_web/urls.py                         |  18 +++++++++++---
 8 files changed, 79 insertions(+), 5 deletions(-)
 create mode 100644 backend/serializers.py

diff --git a/backend/models.py b/backend/models.py
index 855c5ef..e15d115 100644
--- a/backend/models.py
+++ b/backend/models.py
@@ -6,11 +6,18 @@ class hosts(models.Model):
     password = models.CharField(max_length=50)
     port = models.IntegerField(null=True, blank=True)
     cluster = models.CharField(max_length=50)
+    def __str__(self):
+        return self.ipaddress
+    
 
 class users(models.Model):
     user = models.CharField(max_length=50)
     password = models.CharField(max_length=50)
+    def __str__(self):
+        return self.user
 
 class cluster(models.Model):
     name = models.CharField(max_length=50)
+    def __str__(self):
+        return self.name
 
diff --git a/backend/serializers.py b/backend/serializers.py
new file mode 100644
index 0000000..f65b460
--- /dev/null
+++ b/backend/serializers.py
@@ -0,0 +1,14 @@
+from django.contrib.auth.models import Group, User
+from rest_framework import serializers
+
+
+class UserSerializer(serializers.HyperlinkedModelSerializer):
+    class Meta:
+        model = User
+        fields = ['url', 'username', 'email', 'groups']
+
+
+class GroupSerializer(serializers.HyperlinkedModelSerializer):
+    class Meta:
+        model = Group
+        fields = ['url', 'name']
\ No newline at end of file
diff --git a/backend/views.py b/backend/views.py
index 91ea44a..264ff20 100644
--- a/backend/views.py
+++ b/backend/views.py
@@ -1,3 +1,23 @@
-from django.shortcuts import render
+from django.contrib.auth.models import Group, User
+from rest_framework import permissions, viewsets
+from backend.serializers import GroupSerializer, UserSerializer
+from django.shortcuts import render, HttpResponse
 
+
+class UserViewSet(viewsets.ModelViewSet):
+    """
+    API endpoint that allows users to be viewed or edited.
+    """
+    queryset = User.objects.all().order_by('-date_joined')
+    serializer_class = UserSerializer
+    # permission_classes = [permissions.IsAuthenticated]
+
+
+class GroupViewSet(viewsets.ModelViewSet):
+    """
+    API endpoint that allows groups to be viewed or edited.
+    """
+    queryset = Group.objects.all()
+    serializer_class = GroupSerializer
+    # permission_classes = [permissions.IsAuthenticated]
 # Create your views here.
diff --git a/safeguard_web/__pycache__/__init__.cpython-39.pyc b/safeguard_web/__pycache__/__init__.cpython-39.pyc
index e93afb5d9e8b3fac386606106a6f2ffafb73ba01..bc0a3466596b1f4e31c7f6e9d4af346b45abb9e1 100644
GIT binary patch
delta 43
xcmZo+>|o?h<mKgJ00P7B%@et8M0E9w^7BjdixbmQ(@PVJQgq8xlO~4i0s!sT47UIP

delta 39
scmeBRY+>Y1<mKgJ00NEDvWeU_0-E|o`S~UKl{v-vdFi^v=@X-M0j^;Rp#T5?

diff --git a/safeguard_web/__pycache__/settings.cpython-39.pyc b/safeguard_web/__pycache__/settings.cpython-39.pyc
index 51ead38da154322885bb8eaca7c3e3bda9f9280c..e3294b9662e774a2b8b7fc6dc1d74cb52b63fbe6 100644
GIT binary patch
delta 545
zcmY*WJ#W)s5YFqo`D$Y4qiNF6kWyf%shZA+$s}$tq;cifq2d<H%{i`t;>b=F!BVLp
zwhomgrc9k#pyFTf3o0c3020hB;Y}ol-syRy=kB??;l7rBmL*#hmts8LpPf4&3KvqH
ze81Rwc#0`ZdjztW!6nRi#3PtR%8^5Hnt+@eL+O%$h<T8(z!k+J=hEfexjdyVfjpL>
zfGb$RDz0+Wc;1>O`PL$?VV(Qd@b-cm3tsn#OQD2!U>R#0P(}(X92L9^RlEnQm}4>d
z{+A!;gk++ZLVXjYzzN6JFtCrEi66Xz%1v#B6Wev2Ao?hLI)!pu-BtQs(_*bB>Yie?
zx(Z|O<eMYSaB$=t+hKFy2aY}Rn##%8XNwvn4(-qzT+{K}k#n7}UyqLLo}7R(Q(JvQ
zGhbNzfY#GYP46+tD*GC<m~NP=0okW)OYd1~TQhY7iu%54C=1(vgZNzOYHfvgt-jH{
zs>d5~lD~SCY!gAQMQ5ou8z31fGp$`i*;5a6;~5m#fzs)yhQ(CV)Owxhcj|NF0e@sL
zr@z}ff$xuZLVM`ACw73_6X!7cmX<e%yi0Srcp*jOR7fV$BAwBcK<G@E38F9$BK%5L
Jqd)1#{{RXxnU??n

delta 248
zcmcaFvPp<9k(ZZ?0SHtNmZd83PUMqeTr*MIvOb+5g?$lY6n6?o6i*5#kjIk31thr_
zF=jHR@XTRJXN=-a;f>-;;Y;C95l9gPVj&<Dp2L>O7{#9=k|GLJCz2utCdI*|L?&Z8
zOO!y0WRzfvRFqH(OO!AWi=;?LiKfUziKWN}Gib^s8Zk03aJg(|W^7|*WS`u{Bs2LO
z)3nWeEON|@yqk5|T^O1DG{q+ubL35C=3FhM0d#p0i1?+cUzDF;qF<R)oS&DjTbw@m
d45y}008oa7hlg2!orRf&iH((mYqBht69DxUH^l$|

diff --git a/safeguard_web/__pycache__/urls.cpython-39.pyc b/safeguard_web/__pycache__/urls.cpython-39.pyc
index e437a816c6353ed6891d110bf40ff00900254903..84f42b62cf93b7f5644ef629fdb5727f0d74ab63 100644
GIT binary patch
literal 845
zcmY*XJ#W-N5Z$%cA9rW_oLq8HA=)c)SJEH^5)z^TlqeM2oHpKx<BLB|yEZ4Tp`gD1
zkd*ulYI>w?srU<sf>|d(vX;m5X5Q@VzOjhoA;I<a(+<8ng#57J?g|k+$5($tBZ;IX
zsVJw^`cCOoE_W-&S&!Y)t9<VFJ}ZMN<YDi7<)9k!Vek9psET-0#XP12hBAN{qTG?;
z&fy98SL18KQ%JA&<N(6e{<VWFi-w3MJ#ys;=;~l~_}KwhM!+CNpY|vQCKEZ9>CWl#
z9wc&KPIhk2_!Op=`GoMPJUAh<!=HE~GNZ;5vMTDCW7t}>7bYm`tn4IUCQwb+g3^}s
zp@5Bb#yPZ0on1gBmNtfUdnX%_=b%nsA44rcfimN_Yp7qm`xoFY0;GCv(U!ndbxT}k
zrO+C*@j8u-jGtr6wch&unHGj@2Vz|uiLSj^Y$pnIyF6E-f=#0?=bb9GN$v;ngo7Fh
zdMz^8da}v%f;0QU46~Uxvj;qpD^cgoJZtJ!6=w=3Pt52xXtz}^MijFqvBtB6;7nwf
zcyQ$-;?w($=Qo;Y71-qC`^lxgQE2}6-dfooOARb{bVp4dzl3wqmF;n_yzy@)U_u3X
zp>Y;`bc#Z~N8wMP<<ToE*DW$k7cCgp%Q8ZX<HU+ue<A9j?Pkrd*+Ml<yU^ks@=hpu
vw1Kn#Xlaq%V7~rn!lzXuyAqyJEYy$CL^P#Q|D;apgfyXv8==MMZx;LkHu3j3

literal 929
zcmb7D%Z}496m`<HQ!*XICwPO>NSzo)tSUet#E6j)3m%(pGH&dqsbV{_of*m&euiJl
zmKDFi2Cm&iDo99RDVFVApL@=^*YxZxMlj~>4G%^LeRG@R0cYqLeC7d646P6gSXc)l
zT!n#KkLr<#R#AX>#3CN<#=FVqunblwZ2S(T6Z$1YDE{#FH7=CgRAsBFt(3$YrLmzK
zUba-T{FbkAqt%WVHjCr;TaK?=T{qNPuBExgb!9Bps<qr;z%tk>d{gmTli^DM5=wJi
zNx&2b^rq(2aBMhV#u&!d+QuviVX810+fiAnAzh&a;gUScE{Ii4RT$!e38Xh9e)*9K
zz%tADvXzA=?$dhEhZ%m(7^V_eqEXr|G2Un;u-NCcY5KzhO&@>J$2$d1z>z|@3SzhO
zBw6&jwo32?mQ?U3iCZUW{Gz7DoUcI`J3{>%3uU)lJ0h89JkWUsSpUPu#b?yy4rHGG
z!RSgBb;~MQf(bsKljiC?_;clr!9WGd<6`+d#hc#B<-x8)IWlMXK(dcrcvJqF?YdSa
z0Rx9ji97JPNt(vrp?y&rbYsdyC1DXc$3IQ{mb^))-ASH9f8_b<Zcho6SY53(bTH~p
zk9cbLD7wg0mUodO-kly5^DEMK;J=@K+$UNoOZK%<vOG5>`K{@Fn`O<un?4qbwKabl
Qz)drSDY`ccW~159Umge`IRF3v

diff --git a/safeguard_web/settings.py b/safeguard_web/settings.py
index e985665..8fc4341 100644
--- a/safeguard_web/settings.py
+++ b/safeguard_web/settings.py
@@ -32,6 +32,7 @@ ALLOWED_HOSTS = ["*"]
 
 INSTALLED_APPS = [
     'backend.apps.BackendConfig',
+    'rest_framework',
     'rest_framework_swagger',
     'django.contrib.admin',
     'django.contrib.auth',
@@ -45,6 +46,28 @@ REST_FRAMEWORK = {
     'DEFAULT_SCHEMA_CLASS': 'rest_framework.schemas.coreapi.AutoSchema'
 }
 
+SWAGGER_SETTINGS = {
+    # 基础样式
+    'SECURITY_DEFINITIONS': {
+        "basic":{
+            'type': 'basic'
+        }
+    },
+    # 如果需要登录才能够查看接口文档, 登录的链接使用restframework自带的.
+    # 'LOGIN_URL': 'rest_framework:login',
+    # 'LOGOUT_URL': 'rest_framework:logout',
+    # 'DOC_EXPANSION': None,
+    # 'SHOW_REQUEST_HEADERS':True,
+    # 'USE_SESSION_AUTH': True,
+    # 'DOC_EXPANSION': 'list',
+    # 接口文档中方法列表以首字母升序排列
+    'APIS_SORTER': 'alpha',
+    # 如果支持json提交, 则接口文档中包含json输入框
+    'JSON_EDITOR': True,
+    # 方法列表字母排序
+    'OPERATIONS_SORTER': 'alpha',
+    'VALIDATOR_URL': None,
+}
 
 MIDDLEWARE = [
     'django.middleware.security.SecurityMiddleware',
diff --git a/safeguard_web/urls.py b/safeguard_web/urls.py
index fee5494..40da000 100644
--- a/safeguard_web/urls.py
+++ b/safeguard_web/urls.py
@@ -1,11 +1,21 @@
 from django.contrib import admin
-from django.urls import path
-from rest_framework_swagger.views import get_swagger_view
+from django.urls import path,include
+from rest_framework import routers
+from backend import views
+from rest_framework.schemas import get_schema_view
+from rest_framework_swagger.renderers import SwaggerUIRenderer, OpenAPIRenderer
 
-schema_view = get_swagger_view(title='API')
+schema_view = get_schema_view(title='API', renderer_classes=[OpenAPIRenderer, SwaggerUIRenderer])
+
+router = routers.DefaultRouter()
+router.register(r'users',views.UserViewSet)
+router.register(r'groups',views.GroupViewSet)
 
 urlpatterns = [
     path('admin/', admin.site.urls),
-    path('doc/', schema_view)
+    path('',include(router.urls)),
+    path('api-auth/',include('rest_framework.urls',namespace='rest_framework')),
+    path('docs/',schema_view,name='docs'),
 ]
 
+urlpatterns += router.urls
\ No newline at end of file
-- 
Gitee