first commit
chunquansang
1 year ago
parent
6e359ac369
commit
a1ed40a446
@ -0,0 +1,2 @@
|
|||||||
|
import pymysql
|
||||||
|
pymysql.install_as_MySQLdb()
|
||||||
@ -0,0 +1,16 @@
|
|||||||
|
"""
|
||||||
|
ASGI config for TP_API project.
|
||||||
|
|
||||||
|
It exposes the ASGI callable as a module-level variable named ``application``.
|
||||||
|
|
||||||
|
For more information on this file, see
|
||||||
|
https://docs.djangoproject.com/en/3.2/howto/deployment/asgi/
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
|
||||||
|
from django.core.asgi import get_asgi_application
|
||||||
|
|
||||||
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'TP_API.settings')
|
||||||
|
|
||||||
|
application = get_asgi_application()
|
||||||
@ -0,0 +1,263 @@
|
|||||||
|
"""
|
||||||
|
Django settings for TP_API project.
|
||||||
|
|
||||||
|
Generated by 'django-admin startproject' using Django 3.2.19.
|
||||||
|
|
||||||
|
For more information on this file, see
|
||||||
|
https://docs.djangoproject.com/en/3.2/topics/settings/
|
||||||
|
|
||||||
|
For the full list of settings and their values, see
|
||||||
|
https://docs.djangoproject.com/en/3.2/ref/settings/
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
from pathlib import Path
|
||||||
|
from elasticsearch_dsl import connections
|
||||||
|
|
||||||
|
# Build paths inside the project like this: BASE_DIR / 'subdir'.
|
||||||
|
BASE_DIR = Path(__file__).resolve().parent.parent
|
||||||
|
|
||||||
|
|
||||||
|
# Quick-start development settings - unsuitable for production
|
||||||
|
# See https://docs.djangoproject.com/en/3.2/howto/deployment/checklist/
|
||||||
|
|
||||||
|
# SECURITY WARNING: keep the secret key used in production secret!
|
||||||
|
SECRET_KEY = 'django-insecure-kqm6r(-!q-emw5c!!@dc)9l^hm@m#4!)-d7ecq85&u^5leu5)_'
|
||||||
|
|
||||||
|
# SECURITY WARNING: don't run with debug turned on in production!
|
||||||
|
DEBUG = False
|
||||||
|
|
||||||
|
ALLOWED_HOSTS = ["*"]
|
||||||
|
|
||||||
|
|
||||||
|
# Application definition
|
||||||
|
|
||||||
|
INSTALLED_APPS = [
|
||||||
|
'django.contrib.admin',
|
||||||
|
'django.contrib.auth',
|
||||||
|
'django.contrib.contenttypes',
|
||||||
|
'django.contrib.sessions',
|
||||||
|
'django.contrib.messages',
|
||||||
|
'django.contrib.staticfiles',
|
||||||
|
'rest_framework',
|
||||||
|
'django_filters',
|
||||||
|
'corsheaders',
|
||||||
|
'app',
|
||||||
|
'event',
|
||||||
|
'user',
|
||||||
|
'department',
|
||||||
|
]
|
||||||
|
|
||||||
|
MIDDLEWARE = [
|
||||||
|
'django.middleware.security.SecurityMiddleware',
|
||||||
|
'django.contrib.sessions.middleware.SessionMiddleware',
|
||||||
|
'corsheaders.middleware.CorsMiddleware',
|
||||||
|
'django.middleware.common.CommonMiddleware',
|
||||||
|
'django.middleware.csrf.CsrfViewMiddleware',
|
||||||
|
'django.contrib.auth.middleware.AuthenticationMiddleware',
|
||||||
|
'django.contrib.messages.middleware.MessageMiddleware',
|
||||||
|
'django.middleware.clickjacking.XFrameOptionsMiddleware',
|
||||||
|
'mymiddleware.middleware.CheckTokenMiddleware',
|
||||||
|
]
|
||||||
|
|
||||||
|
ROOT_URLCONF = 'TP_API.urls'
|
||||||
|
|
||||||
|
TEMPLATES = [
|
||||||
|
{
|
||||||
|
'BACKEND': 'django.template.backends.django.DjangoTemplates',
|
||||||
|
'DIRS': [],
|
||||||
|
'APP_DIRS': True,
|
||||||
|
'OPTIONS': {
|
||||||
|
'context_processors': [
|
||||||
|
'django.template.context_processors.debug',
|
||||||
|
'django.template.context_processors.request',
|
||||||
|
'django.contrib.auth.context_processors.auth',
|
||||||
|
'django.contrib.messages.context_processors.messages',
|
||||||
|
],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
WSGI_APPLICATION = 'TP_API.wsgi.application'
|
||||||
|
|
||||||
|
|
||||||
|
# Database
|
||||||
|
# https://docs.djangoproject.com/en/3.2/ref/settings/#databases
|
||||||
|
|
||||||
|
# DATABASES = {
|
||||||
|
# 'default': {
|
||||||
|
# 'ENGINE': 'django.db.backends.sqlite3',
|
||||||
|
# 'NAME': BASE_DIR / 'db.sqlite3',
|
||||||
|
# }
|
||||||
|
# }
|
||||||
|
|
||||||
|
DATABASES = {
|
||||||
|
'default': {
|
||||||
|
'ENGINE': 'django.db.backends.mysql',
|
||||||
|
'NAME': os.getenv("MYSQL_DATABASE") or 'tp_11_23',
|
||||||
|
'USER': os.getenv("MYSQL_USER") or 'root',
|
||||||
|
'PASSWORD': os.getenv("MYSQL_PASSWORD") or '#Yaxin0504',
|
||||||
|
'HOST': os.getenv("MYSQL_HOST") or '192.168.10.96',
|
||||||
|
'PORT': os.getenv("MYSQL_PORT") or '3306',
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# Password validation
|
||||||
|
# https://docs.djangoproject.com/en/3.2/ref/settings/#auth-password-validators
|
||||||
|
|
||||||
|
AUTH_PASSWORD_VALIDATORS = [
|
||||||
|
{
|
||||||
|
'NAME': 'django.contrib.auth.password_validation.UserAttributeSimilarityValidator',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
'NAME': 'django.contrib.auth.password_validation.MinimumLengthValidator',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
'NAME': 'django.contrib.auth.password_validation.CommonPasswordValidator',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
'NAME': 'django.contrib.auth.password_validation.NumericPasswordValidator',
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
# Internationalization
|
||||||
|
# https://docs.djangoproject.com/en/3.2/topics/i18n/
|
||||||
|
|
||||||
|
LANGUAGE_CODE = 'zh-hans'
|
||||||
|
|
||||||
|
TIME_ZONE = 'Asia/Shanghai'
|
||||||
|
|
||||||
|
USE_I18N = True
|
||||||
|
|
||||||
|
USE_L10N = True
|
||||||
|
|
||||||
|
USE_TZ = False
|
||||||
|
|
||||||
|
|
||||||
|
# Static files (CSS, JavaScript, Images)
|
||||||
|
# https://docs.djangoproject.com/en/3.2/howto/static-files/
|
||||||
|
|
||||||
|
STATIC_URL = '/static/'
|
||||||
|
|
||||||
|
STATICFILES_DIRS = [
|
||||||
|
os.path.join(BASE_DIR, 'static'),
|
||||||
|
]
|
||||||
|
|
||||||
|
MEDIA_URL = '/media/'
|
||||||
|
MEDIA_ROOT = os.path.join(BASE_DIR, 'media')
|
||||||
|
|
||||||
|
# Default primary key field type
|
||||||
|
# https://docs.djangoproject.com/en/3.2/ref/settings/#default-auto-field
|
||||||
|
|
||||||
|
DEFAULT_AUTO_FIELD = 'django.db.models.BigAutoField'
|
||||||
|
|
||||||
|
# 解决跨域
|
||||||
|
CORS_ALLOW_CREDENTIALS = True
|
||||||
|
CORS_ORIGIN_ALLOW_ALL = True
|
||||||
|
CORS_ALLOW_HEADERS = ["*"]
|
||||||
|
|
||||||
|
# CORS_ORIGIN_WHITELIST = ('*',)
|
||||||
|
# 对应的发送的请求的跨域
|
||||||
|
CORS_ALLOW_METHODS = (
|
||||||
|
'DELETE',
|
||||||
|
'GET',
|
||||||
|
'OPTIONS',
|
||||||
|
'PATCH',
|
||||||
|
'POST',
|
||||||
|
'PUT',
|
||||||
|
'VIEW',
|
||||||
|
)
|
||||||
|
|
||||||
|
USE_L10N = False
|
||||||
|
DATE_FORMAT = 'Y-m-d'
|
||||||
|
DATETIME_FORMAT = 'Y-m-d H:M:S'
|
||||||
|
|
||||||
|
REST_FRAMEWORK = {
|
||||||
|
'DATETIME_FORMAT': "%Y-%m-%d %H:%M:%S",
|
||||||
|
'DATE_FORMAT': "%Y-%m-%d",
|
||||||
|
'DEFAULT_FILTER_BACKENDS': ['django_filters.rest_framework.DjangoFilterBackend'],
|
||||||
|
'EXCEPTION_HANDLER': 'app.exception.exception_handler',
|
||||||
|
# 'DEFAULT_AUTHENTICATION_CLASSES': (
|
||||||
|
# 'rest_framework_jwt.authentication.JSONWebTokenAuthentication', # 第一种jwt方式
|
||||||
|
# # 'rest_framework.authentication.SessionAuthentication', # 第二种session方式
|
||||||
|
# # 'rest_framework.authentication.BasicAuthentication', # 第三种Django的基本方式
|
||||||
|
# ),
|
||||||
|
#
|
||||||
|
# 'DEFAULT_PERMISSION_CLASSES': (
|
||||||
|
# 'rest_framework.permissions.IsAuthenticated',
|
||||||
|
# ),
|
||||||
|
}
|
||||||
|
|
||||||
|
# 配置日志
|
||||||
|
LOG_DIR = os.path.join(BASE_DIR, 'logs')
|
||||||
|
LOGGING = {
|
||||||
|
'version': 1, # 保留字
|
||||||
|
'disable_existing_loggers': False, # 是否禁用已经存在的日志实例
|
||||||
|
'formatters': { # 定义日志的格式
|
||||||
|
'standard': {
|
||||||
|
'format': '[%(asctime)s][%(threadName)s:%(thread)d][task_id:%(name)s][%(filename)s:%(lineno)d]'
|
||||||
|
'[%(levelname)s]%(message)s'
|
||||||
|
},
|
||||||
|
'simple': {
|
||||||
|
'format': '[%(levelname)s][%(asctime)s][%(filename)s:%(lineno)d]%(message)s'
|
||||||
|
},
|
||||||
|
'collect': {
|
||||||
|
'format': '%(message)s'
|
||||||
|
}
|
||||||
|
},
|
||||||
|
'filters': { # 定义日志的过滤器
|
||||||
|
'require_debug_true': {
|
||||||
|
'()': 'django.utils.log.RequireDebugTrue',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
'handlers': { # 日志处理程序
|
||||||
|
'console': {
|
||||||
|
'level': 'DEBUG',
|
||||||
|
# 'filters': ['require_debug_true'], # 只有在Django debug为True时才在屏幕打印日志
|
||||||
|
'class': 'logging.StreamHandler',
|
||||||
|
'formatter': 'simple',
|
||||||
|
# 'filename': os.path.join(BASE_LOG_DIR, "tpservice.log")
|
||||||
|
},
|
||||||
|
'file': {
|
||||||
|
'level': 'INFO',
|
||||||
|
'class': 'logging.handlers.RotatingFileHandler', # window多进程 需要 pip install concurrent-log-handler
|
||||||
|
'filename': os.path.join(BASE_DIR, "logs/debug.log"), # 日志文件
|
||||||
|
'backupCount': 10, # 保留的最大文件数，超过则删除日期最早的
|
||||||
|
'maxBytes': 1024 * 1024 * 100, # 文件大小
|
||||||
|
'formatter': 'standard',
|
||||||
|
'encoding': 'utf-8',
|
||||||
|
},
|
||||||
|
# 'file': {
|
||||||
|
# 'level': 'INFO',
|
||||||
|
# 'class': 'cloghandler.ConcurrentRotatingFileHandler', # linux多进程 需要 pip install ConcurrentLogHandler
|
||||||
|
# 'filename': os.path.join(BASE_DIR, "logs/debug.log"), # 日志文件
|
||||||
|
# 'backupCount': 10, # 保留的最大文件数，超过则删除日期最早的
|
||||||
|
# 'maxBytes': 1024 * 1024 * 10, # 文件大小
|
||||||
|
# 'formatter': 'standard',
|
||||||
|
# 'encoding': 'utf-8',
|
||||||
|
# },
|
||||||
|
},
|
||||||
|
'loggers': { # 日志实例 记录器
|
||||||
|
'mylogger': { # 默认的logger应用如下配置
|
||||||
|
'handlers': ['console', 'file'],
|
||||||
|
'level': 'DEBUG',
|
||||||
|
'propagate': True, # 是否向上一级logger实例传递日志信息
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
AUTH_USER_MODEL = 'user.UserProfile'
|
||||||
|
|
||||||
|
# connections.create_connection(
|
||||||
|
# hosts=['192.168.10.96:9208'],
|
||||||
|
# http_auth=('elastic', 'changeme')
|
||||||
|
# )
|
||||||
|
ES_USER = os.getenv("ES_USER") or "elastic"
|
||||||
|
ES_PASSWORD = os.getenv("ES_PASSWORD") or "changeme"
|
||||||
|
ES_IP = os.getenv("ES_IP") or "192.168.10.96:9209"
|
||||||
|
connections.create_connection(hosts=[f"http://{ES_USER}:{ES_PASSWORD}@{ES_IP}"])
|
||||||
|
|
||||||
|
VIDEO_DOWNLOAD_PATH = os.getenv("VIDEO_DOWNLOAD_PATH") or '/home/tp/dsr/TP-Code-AI/src/media/videos'
|
||||||
|
|
||||||
@ -0,0 +1,35 @@
|
|||||||
|
"""TP_Api URL Configuration
|
||||||
|
|
||||||
|
The `urlpatterns` list routes URLs to views. For more information please see:
|
||||||
|
https://docs.djangoproject.com/en/3.2/topics/http/urls/
|
||||||
|
Examples:
|
||||||
|
Function views
|
||||||
|
1. Add an import: from my_app import views
|
||||||
|
2. Add a URL to urlpatterns: path('', views.home, name='home')
|
||||||
|
Class-based views
|
||||||
|
1. Add an import: from other_app.views import Home
|
||||||
|
2. Add a URL to urlpatterns: path('', Home.as_view(), name='home')
|
||||||
|
Including another URLconf
|
||||||
|
1. Import the include() function: from django.urls import include, path
|
||||||
|
2. Add a URL to urlpatterns: path('blog/', include('blog.urls'))
|
||||||
|
"""
|
||||||
|
from django.contrib import admin
|
||||||
|
from django.urls import path, include, re_path
|
||||||
|
from django.views.static import serve
|
||||||
|
from django.conf.urls.static import static
|
||||||
|
from django.conf import settings
|
||||||
|
from .settings import MEDIA_ROOT
|
||||||
|
|
||||||
|
|
||||||
|
urlpatterns = [
|
||||||
|
path('admin/', admin.site.urls),
|
||||||
|
path('api/tps/', include('app.urls')),
|
||||||
|
path('event/', include('event.urls')),
|
||||||
|
path('api/user/', include('user.urls')),
|
||||||
|
path('api/departments/', include('department.urls')),
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
urlpatterns += [
|
||||||
|
re_path(r'^media/(?P<path>.*)$', serve, {'document_root': MEDIA_ROOT}),
|
||||||
|
]
|
||||||
@ -0,0 +1,16 @@
|
|||||||
|
"""
|
||||||
|
WSGI config for TP_API project.
|
||||||
|
|
||||||
|
It exposes the WSGI callable as a module-level variable named ``application``.
|
||||||
|
|
||||||
|
For more information on this file, see
|
||||||
|
https://docs.djangoproject.com/en/3.2/howto/deployment/wsgi/
|
||||||
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
|
||||||
|
from django.core.wsgi import get_wsgi_application
|
||||||
|
|
||||||
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'TP_API.settings')
|
||||||
|
|
||||||
|
application = get_wsgi_application()
|
||||||
@ -0,0 +1,5 @@
|
|||||||
|
from django.contrib import admin
|
||||||
|
|
||||||
|
# Register your models here.
|
||||||
|
# from app.models import TP
|
||||||
|
# admin.site.register(TP)
|
||||||
@ -0,0 +1,6 @@
|
|||||||
|
from django.apps import AppConfig
|
||||||
|
|
||||||
|
|
||||||
|
class App01Config(AppConfig):
|
||||||
|
default_auto_field = 'django.db.models.BigAutoField'
|
||||||
|
name = 'app'
|
||||||
@ -0,0 +1,59 @@
|
|||||||
|
from elasticsearch_dsl import Document, Date, Text, Integer, Nested, Keyword, Object
|
||||||
|
|
||||||
|
|
||||||
|
class VideoIndex(Document):
|
||||||
|
STATUS = ((0, "未分析"), (1, "分析中"), (2, "分析完成"), (3, "分析失败"))
|
||||||
|
DOWNLOAD_STATUS = ((0, "未下载"), (1, "下载中"), (2, "下载完成"))
|
||||||
|
|
||||||
|
video_name = Text(fields={'keyword': {'type': 'keyword'}}) # 添加一个 keyword 字段以支持精确匹配
|
||||||
|
video_path = Text(fields={'keyword': {'type': 'keyword'}})
|
||||||
|
ftp_path = Text(fields={'keyword': {'type': 'keyword'}})
|
||||||
|
status = Integer()
|
||||||
|
download_status = Integer()
|
||||||
|
create_time = Date()
|
||||||
|
update_time = Date()
|
||||||
|
video_date = Date()
|
||||||
|
video_hash = Text(fields={'keyword': {'type': 'keyword'}})
|
||||||
|
video_status = Integer()
|
||||||
|
|
||||||
|
class Index:
|
||||||
|
name = 'video_index' # 索引名称
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = "tp_videos"
|
||||||
|
|
||||||
|
|
||||||
|
class TaskIndex(Document):
|
||||||
|
STATUS = ((0, "未分析"), (1, "分析中"), (2, "分析完成"), (3, "分析失败"))
|
||||||
|
|
||||||
|
taskId = Integer()
|
||||||
|
analysis_status = Integer()
|
||||||
|
create_time = Date()
|
||||||
|
update_time = Date()
|
||||||
|
error_reason = Text(fields={'keyword': {'type': 'keyword'}})
|
||||||
|
|
||||||
|
video_id = Integer()
|
||||||
|
|
||||||
|
class Index:
|
||||||
|
name = 'task_index'
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = 'tp_tasks'
|
||||||
|
|
||||||
|
|
||||||
|
class TPResIndex(Document):
|
||||||
|
relative_time = Date()
|
||||||
|
exception_type = Text()
|
||||||
|
abnormal_id = Text()
|
||||||
|
abnormal_behavior = Text(fields={'keyword': {'type': 'keyword'}})
|
||||||
|
abnormal_pic = Text()
|
||||||
|
abnormal_video = Text()
|
||||||
|
video_id = Integer()
|
||||||
|
create_time = Date()
|
||||||
|
update_time = Date()
|
||||||
|
|
||||||
|
class Index:
|
||||||
|
name = 'tpres_index'
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = 'tp_res'
|
||||||
@ -0,0 +1,23 @@
|
|||||||
|
import logging
|
||||||
|
from rest_framework.views import exception_handler as drf_exception_handler # drf原生处理异常函数取别名
|
||||||
|
from rest_framework.views import Response
|
||||||
|
from rest_framework import status
|
||||||
|
|
||||||
|
|
||||||
|
logger = logging.getLogger('mylogger')
|
||||||
|
|
||||||
|
|
||||||
|
def exception_handler(exc, context):
|
||||||
|
# drf的exception_handler做基础处理
|
||||||
|
response = drf_exception_handler(exc, context)
|
||||||
|
# 为空，进行自定义二次处理
|
||||||
|
logger.error(str(exc))
|
||||||
|
if response is None:
|
||||||
|
# print(exc) # 错误原因
|
||||||
|
# print(context) # 错误信息
|
||||||
|
# print('%s - %s - %s' % (context['view'], context['request'].method, exc))
|
||||||
|
return Response({
|
||||||
|
'detail': '服务器错误'
|
||||||
|
}, status=status.HTTP_500_INTERNAL_SERVER_ERROR, exception=True)
|
||||||
|
return response
|
||||||
|
|
||||||
@ -0,0 +1,67 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-09-22 09:47
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
import django.db.models.deletion
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
initial = True
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='Video',
|
||||||
|
fields=[
|
||||||
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('video_name', models.CharField(max_length=512, null=True, verbose_name='视频名称')),
|
||||||
|
('video_path', models.CharField(max_length=512, null=True, verbose_name='视频路径')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'db_table': 'tp_videos',
|
||||||
|
},
|
||||||
|
),
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='TPRes',
|
||||||
|
fields=[
|
||||||
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('relative_time', models.FloatField(blank=True, null=True, verbose_name='相对时间')),
|
||||||
|
('exception_type', models.CharField(max_length=128, null=True, verbose_name='异常类型')),
|
||||||
|
('abnormal_id', models.CharField(max_length=256, null=True, verbose_name='异常ID')),
|
||||||
|
('abnormal_behavior', models.CharField(max_length=256, null=True, verbose_name='异常行为')),
|
||||||
|
('abnormal_pic', models.CharField(max_length=1024, null=True, verbose_name='异常关键帧/图片')),
|
||||||
|
('abnormal_video', models.CharField(max_length=1024, null=True, verbose_name='异常视频定位/视频')),
|
||||||
|
('video', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='app.video')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'db_table': 'tp_tp',
|
||||||
|
},
|
||||||
|
),
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='Person',
|
||||||
|
fields=[
|
||||||
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('pic_path', models.CharField(max_length=512, null=True, verbose_name='图片路径')),
|
||||||
|
('pic_quality', models.IntegerField(null=True, verbose_name='图片质量')),
|
||||||
|
('video', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='app.video')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'db_table': 'tp_person',
|
||||||
|
},
|
||||||
|
),
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='Car',
|
||||||
|
fields=[
|
||||||
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('pic_path', models.CharField(max_length=512, null=True, verbose_name='图片路径')),
|
||||||
|
('pic_quality', models.IntegerField(null=True, verbose_name='图片质量')),
|
||||||
|
('car_number', models.CharField(max_length=512, null=True, verbose_name='车牌号')),
|
||||||
|
('video', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='app.video')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'db_table': 'tp_car',
|
||||||
|
},
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,18 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-09-26 14:08
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0001_initial'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='video',
|
||||||
|
name='status',
|
||||||
|
field=models.IntegerField(choices=[(0, '未分析'), (1, '分析中'), (2, '分析完成')], default=0, verbose_name='视频状态'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,23 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-09-28 09:52
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0002_video_status'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='video',
|
||||||
|
name='create_time',
|
||||||
|
field=models.DateTimeField(auto_now_add=True, null=True, verbose_name='创建时间'),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='video',
|
||||||
|
name='update_time',
|
||||||
|
field=models.DateTimeField(auto_now=True, null=True, verbose_name='更新时间'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,43 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-10-12 15:12
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0003_auto_20230928_0952'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='video',
|
||||||
|
name='download_status',
|
||||||
|
field=models.IntegerField(choices=[(0, '未下载'), (1, '下载中'), (2, '下载完成')], default=0, verbose_name='视频下载状态'),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='video',
|
||||||
|
name='ftp_path',
|
||||||
|
field=models.CharField(max_length=512, null=True, verbose_name='视频ftp路径'),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='video',
|
||||||
|
name='video_date',
|
||||||
|
field=models.CharField(max_length=512, null=True, verbose_name='视频所属日期'),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='video',
|
||||||
|
name='video_hash',
|
||||||
|
field=models.CharField(max_length=512, null=True, unique=True, verbose_name='视频哈希'),
|
||||||
|
),
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='video',
|
||||||
|
name='status',
|
||||||
|
field=models.IntegerField(choices=[(0, '未分析'), (1, '分析中'), (2, '分析完成')], default=0, verbose_name='视频分析状态'),
|
||||||
|
),
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='video',
|
||||||
|
name='video_path',
|
||||||
|
field=models.CharField(max_length=512, null=True, verbose_name='视频本地路径'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,23 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-10-17 10:22
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0004_auto_20231012_1512'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='video',
|
||||||
|
name='status',
|
||||||
|
field=models.IntegerField(choices=[(0, '未分析'), (1, '分析中'), (2, '分析完成'), (3, '分析失败')], default=0, verbose_name='视频分析状态'),
|
||||||
|
),
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='video',
|
||||||
|
name='video_date',
|
||||||
|
field=models.DateField(max_length=512, null=True, verbose_name='视频所属日期'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,18 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-10-17 16:39
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0005_auto_20231017_1022'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='video',
|
||||||
|
name='error_reason',
|
||||||
|
field=models.CharField(max_length=512, null=True, unique=True, verbose_name='視頻哈希'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,18 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-10-17 17:13
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0006_video_error_reason'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='video',
|
||||||
|
name='error_reason',
|
||||||
|
field=models.CharField(max_length=512, null=True, verbose_name='視頻哈希'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,18 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-10-25 13:33
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0007_alter_video_error_reason'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='video',
|
||||||
|
name='ftp_path',
|
||||||
|
field=models.CharField(max_length=512, null=True, unique=True, verbose_name='视频ftp路径'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,23 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-10-30 11:36
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0008_alter_video_ftp_path'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='car',
|
||||||
|
name='pic_quality',
|
||||||
|
field=models.CharField(max_length=128, null=True, verbose_name='图片质量'),
|
||||||
|
),
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='person',
|
||||||
|
name='pic_quality',
|
||||||
|
field=models.CharField(max_length=128, null=True, verbose_name='图片质量'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,27 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-11-08 13:57
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
import django.db.models.deletion
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0009_auto_20231030_1136'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='Task',
|
||||||
|
fields=[
|
||||||
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('analysis_status', models.IntegerField(choices=[(0, '未分析'), (1, '分析中'), (2, '分析完成'), (3, '分析失败')], default=0, verbose_name='视频分析状态')),
|
||||||
|
('create_time', models.DateTimeField(auto_now_add=True, null=True, verbose_name='创建时间')),
|
||||||
|
('update_time', models.DateTimeField(auto_now=True, null=True, verbose_name='更新时间')),
|
||||||
|
('video', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='app.video')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'db_table': 'tp_tasks',
|
||||||
|
},
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,22 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-11-09 11:09
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0010_task'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.RemoveField(
|
||||||
|
model_name='task',
|
||||||
|
name='id',
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='task',
|
||||||
|
name='taskId',
|
||||||
|
field=models.AutoField(default=None, primary_key=True, serialize=False),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,22 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-11-09 11:17
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0011_auto_20231109_1109'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.RemoveField(
|
||||||
|
model_name='video',
|
||||||
|
name='error_reason',
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='task',
|
||||||
|
name='error_reason',
|
||||||
|
field=models.CharField(max_length=512, null=True, verbose_name='错误信息'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,18 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-11-17 15:24
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0012_auto_20231109_1117'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='video',
|
||||||
|
name='video_status',
|
||||||
|
field=models.IntegerField(choices=[(1, '有效视频'), (2, '无效视频'), (3, '异常视频')], default=1, verbose_name='视频有效/无效'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,18 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-11-24 10:16
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0013_video_video_status'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='task',
|
||||||
|
name='callbackUrl',
|
||||||
|
field=models.CharField(max_length=512, null=True, verbose_name='回调地址'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,18 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-12-13 14:18
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0014_task_callbackurl'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='task',
|
||||||
|
name='analysis_status',
|
||||||
|
field=models.IntegerField(choices=[(100, '未分析'), (101, '分析中'), (102, '分析完成'), (103, "Couldn't open webcam or video"), (104, '视频文件不存在'), (105, '切分视频失败'), (106, '未知异常')], default=100, verbose_name='视频分析状态'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,28 @@
|
|||||||
|
# Generated by Django 3.2.25 on 2024-03-14 16:27
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('app', '0015_alter_task_analysis_status'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='video',
|
||||||
|
name='download_status',
|
||||||
|
field=models.IntegerField(choices=[(0, '未下载'), (1, '下载中'), (2, '下载完成'), (3, '下载异常')], default=0, verbose_name='视频下载状态'),
|
||||||
|
),
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='video',
|
||||||
|
name='video_hash',
|
||||||
|
field=models.CharField(max_length=512, null=True, verbose_name='视频哈希'),
|
||||||
|
),
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='video',
|
||||||
|
name='video_status',
|
||||||
|
field=models.IntegerField(choices=[(1, '有效视频'), (2, '无效视频'), (3, '异常视频'), (4, '重复视频')], default=1, verbose_name='视频有效/无效'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,112 @@
|
|||||||
|
from django.db import models
|
||||||
|
|
||||||
|
# Create your models here.
|
||||||
|
|
||||||
|
|
||||||
|
class Video(models.Model):
|
||||||
|
STATUS = ((0, "未分析"), (1, "分析中"), (2, "分析完成"), (3, "分析失败"))
|
||||||
|
DOWNLOAD_STATUS = ((0, "未下载"), (1, "下载中"), (2, "下载完成"), (3, "下载异常"))
|
||||||
|
VIDEO_STATUS = ((1, "有效视频"), (2, "无效视频"), (3, "异常视频"), (4, "重复视频"))
|
||||||
|
video_name = models.CharField(max_length=512, verbose_name='视频名称', null=True)
|
||||||
|
video_path = models.CharField(max_length=512, verbose_name='视频本地路径', null=True)
|
||||||
|
ftp_path = models.CharField(max_length=512, verbose_name='视频ftp路径', null=True, unique=True)
|
||||||
|
status = models.IntegerField(verbose_name='视频分析状态', choices=STATUS, default=0)
|
||||||
|
download_status = models.IntegerField(verbose_name='视频下载状态', default=0, choices=DOWNLOAD_STATUS)
|
||||||
|
create_time = models.DateTimeField(auto_now_add=True, verbose_name='创建时间', null=True)
|
||||||
|
update_time = models.DateTimeField(auto_now=True, verbose_name='更新时间', null=True)
|
||||||
|
video_date = models.DateField(verbose_name='视频所属日期', max_length=512, null=True)
|
||||||
|
video_hash = models.CharField(verbose_name='视频哈希', max_length=512, null=True)
|
||||||
|
video_status = models.IntegerField(verbose_name='视频有效/无效', choices=VIDEO_STATUS, default=1)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = "tp_videos"
|
||||||
|
|
||||||
|
|
||||||
|
class Task(models.Model):
|
||||||
|
"""任务表 与视频表一对多？"""
|
||||||
|
STATUS = ((100, "未分析"), (101, "分析中"), (102, "分析完成"), (103, "Couldn't open webcam or video"), (104, '视频文件不存在'), (105, '切分视频失败'), (106, '未知异常'))
|
||||||
|
taskId = models.AutoField(primary_key=True, default=None)
|
||||||
|
analysis_status = models.IntegerField(verbose_name='视频分析状态', choices=STATUS, default=100)
|
||||||
|
create_time = models.DateTimeField(auto_now_add=True, verbose_name='创建时间', null=True)
|
||||||
|
update_time = models.DateTimeField(auto_now=True, verbose_name='更新时间', null=True)
|
||||||
|
video = models.ForeignKey(Video, on_delete=models.CASCADE, null=True)
|
||||||
|
error_reason = models.CharField(max_length=512, verbose_name='错误信息', null=True)
|
||||||
|
callbackUrl = models.CharField(max_length=512, verbose_name='回调地址', null=True)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = 'tp_tasks'
|
||||||
|
|
||||||
|
# class TP(models.Model):
|
||||||
|
# # uid
|
||||||
|
# uid = models.AutoField(primary_key=True)
|
||||||
|
# # 视频哈希
|
||||||
|
# video_hash = models.CharField(max_length=1024, verbose_name='视频哈希')
|
||||||
|
# # 记录仪时间
|
||||||
|
# record_time = models.DateTimeField(verbose_name='记录仪时间', null=True, blank=True)
|
||||||
|
# # 设备号
|
||||||
|
# equipment_id = models.CharField(max_length=64, null=True, blank=True, verbose_name='设备号')
|
||||||
|
# # 警号
|
||||||
|
# police_id = models.CharField(max_length=50, null=True, blank=True, verbose_name='警号') # 警号可为空可不传
|
||||||
|
# # 事件类型
|
||||||
|
# event_type = models.CharField(max_length=50, verbose_name='事件类型/车辆违法原因', null=True, blank=True) # 还没迁移
|
||||||
|
# # 是否违规
|
||||||
|
# is_violation = models.BooleanField(verbose_name='执法人员是否违规', null=True, blank=True)
|
||||||
|
# # 缩略图
|
||||||
|
# small_image = models.CharField(max_length=1024, verbose_name='缩略图', null=True, blank=True)
|
||||||
|
# # 相对时间
|
||||||
|
# relative_time = models.FloatField(verbose_name='相对时间', null=True, blank=True)
|
||||||
|
# # 视频路径
|
||||||
|
# video_dir = models.CharField(max_length=1024, verbose_name='视频路径')
|
||||||
|
# # 车牌号
|
||||||
|
# car_number = models.CharField(max_length=1024, verbose_name='车牌号', null=True, blank=True) # 车牌可为空可不传
|
||||||
|
# # 分析结果
|
||||||
|
# ai_analysis = models.CharField(max_length=255, verbose_name='执法人员违规行为', null=True, blank=True) # 分析结果可为空可不传
|
||||||
|
# # 加入时间
|
||||||
|
# add_time = models.DateTimeField(auto_now_add=True, verbose_name='加入时间')
|
||||||
|
# # 更新时间
|
||||||
|
# update_time = models.DateTimeField(auto_now=True, verbose_name='更新时间')
|
||||||
|
# # 是否显示
|
||||||
|
# is_display = models.BooleanField(default=True, verbose_name='是否显示')
|
||||||
|
#
|
||||||
|
# is_illegal = models.BooleanField(verbose_name='车辆是否违法', null=True, blank=True)
|
||||||
|
#
|
||||||
|
# video = models.ForeignKey(Video, on_delete=models.CASCADE, null=True)
|
||||||
|
#
|
||||||
|
# class Meta:
|
||||||
|
# db_table = "app_tp"
|
||||||
|
#
|
||||||
|
# # 排序 uid倒序
|
||||||
|
# ordering = ['-uid']
|
||||||
|
|
||||||
|
|
||||||
|
class Person(models.Model):
|
||||||
|
pic_path = models.CharField(max_length=512, verbose_name='图片路径', null=True)
|
||||||
|
pic_quality = models.CharField(verbose_name='图片质量', null=True, max_length=128)
|
||||||
|
video = models.ForeignKey(Video, on_delete=models.CASCADE, null=True)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = "tp_person"
|
||||||
|
|
||||||
|
|
||||||
|
class Car(models.Model):
|
||||||
|
pic_path = models.CharField(max_length=512, verbose_name='图片路径', null=True)
|
||||||
|
pic_quality = models.CharField(verbose_name='图片质量', null=True, max_length=128)
|
||||||
|
car_number = models.CharField(max_length=512, verbose_name='车牌号', null=True)
|
||||||
|
video = models.ForeignKey(Video, on_delete=models.CASCADE, null=True)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = "tp_car"
|
||||||
|
|
||||||
|
|
||||||
|
# 视频内时间、异常类型、异常ID、异常行为、异常关键帧、异常视频定位
|
||||||
|
class TPRes(models.Model):
|
||||||
|
relative_time = models.FloatField(verbose_name='相对时间', null=True, blank=True)
|
||||||
|
exception_type = models.CharField(max_length=128, verbose_name='异常类型', null=True)
|
||||||
|
abnormal_id = models.CharField(max_length=256, verbose_name='异常ID', null=True)
|
||||||
|
abnormal_behavior = models.CharField(max_length=256, verbose_name='异常行为', null=True)
|
||||||
|
abnormal_pic = models.CharField(max_length=1024, verbose_name='异常关键帧/图片', null=True)
|
||||||
|
abnormal_video = models.CharField(max_length=1024, verbose_name='异常视频定位/视频', null=True)
|
||||||
|
video = models.ForeignKey(Video, on_delete=models.CASCADE, null=True)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = "tp_tp"
|
||||||
@ -0,0 +1,8 @@
|
|||||||
|
from rest_framework.pagination import PageNumberPagination
|
||||||
|
|
||||||
|
|
||||||
|
class MyPageNumberPagination(PageNumberPagination):
|
||||||
|
page_size = 10
|
||||||
|
page_query_param = "page"
|
||||||
|
page_size_query_param = "page_size"
|
||||||
|
max_page_size = 100
|
||||||
@ -0,0 +1,3 @@
|
|||||||
|
from django.test import TestCase
|
||||||
|
|
||||||
|
# Create your tests here.
|
||||||
@ -0,0 +1,39 @@
|
|||||||
|
"""TP_Api URL Configuration
|
||||||
|
|
||||||
|
The `urlpatterns` list routes URLs to views. For more information please see:
|
||||||
|
https://docs.djangoproject.com/en/3.2/topics/http/urls/
|
||||||
|
Examples:
|
||||||
|
Function views
|
||||||
|
1. Add an import: from my_app import views
|
||||||
|
2. Add a URL to urlpatterns: path('', views.home, name='home')
|
||||||
|
Class-based views
|
||||||
|
1. Add an import: from other_app.views import Home
|
||||||
|
2. Add a URL to urlpatterns: path('', Home.as_view(), name='home')
|
||||||
|
Including another URLconf
|
||||||
|
1. Import the include() function: from django.urls import include, path
|
||||||
|
2. Add a URL to urlpatterns: path('blog/', include('blog.urls'))
|
||||||
|
"""
|
||||||
|
# from django.contrib import admin
|
||||||
|
from django.urls import path, include, re_path
|
||||||
|
from rest_framework.routers import DefaultRouter
|
||||||
|
from app import views
|
||||||
|
|
||||||
|
# router = DefaultRouter()
|
||||||
|
# router.register('', views.ModelQuery)
|
||||||
|
|
||||||
|
urlpatterns = [
|
||||||
|
# path('', include(router.urls)),
|
||||||
|
# path('events', views.ModelQuery.as_view({"get": "query_event"})),
|
||||||
|
# path('', views.TPAPIView.as_view()),
|
||||||
|
# path('<pk>/', views.TPAPIView.as_view()),
|
||||||
|
path('upload_video/', views.VideoProcess.as_view({"post": "create", 'get': 'list'})),
|
||||||
|
path('analysis_video/', views.AnalysisVideo.as_view({"post": "create"})),
|
||||||
|
path('pics/', views.PicModelViewSet.as_view({'get': 'list'})),
|
||||||
|
path('', views.TPModelViewSet.as_view({'get': 'list', 'post': 'create'})),
|
||||||
|
path('get_analyzed_videos/', views.GetAnalyzedVideo.as_view({'post': 'list'})),
|
||||||
|
path('get_unanalyzed_videos/', views.VideoAnalysis.as_view({'get': 'list'})),
|
||||||
|
path('update_video_status/', views.UpdateVideoStatus.as_view({'post': 'update'})),
|
||||||
|
path('tasks/', views.TaskViewSet.as_view({'get': 'list', 'post': 'update'})),
|
||||||
|
path('download_video/', views.DownloadVideo.as_view({'post': 'create'})),
|
||||||
|
path('get_video_result/', views.VideoResult.as_view({'get': 'list'})),
|
||||||
|
]
|
||||||
@ -0,0 +1,73 @@
|
|||||||
|
import time
|
||||||
|
import datetime
|
||||||
|
import jwt
|
||||||
|
import hashlib
|
||||||
|
from os import path
|
||||||
|
from django.conf import settings
|
||||||
|
from rest_framework.response import Response
|
||||||
|
from django.contrib.auth.models import User
|
||||||
|
|
||||||
|
|
||||||
|
def generate_token(user):
|
||||||
|
"""
|
||||||
|
:param user: 用户对象
|
||||||
|
:return: 生成的token
|
||||||
|
"""
|
||||||
|
payload = {
|
||||||
|
'user_id': user.id,
|
||||||
|
'username': user.username,
|
||||||
|
'exp': datetime.datetime.utcnow() + datetime.timedelta(weeks=1) # token过期时间 1week
|
||||||
|
}
|
||||||
|
token = jwt.encode(payload=payload, key=settings.SECRET_KEY, algorithm='HS256')
|
||||||
|
return token
|
||||||
|
|
||||||
|
|
||||||
|
def decode_token_exp_time(token):
|
||||||
|
try:
|
||||||
|
res_dict = jwt.decode(token, settings.SECRET_KEY, algorithms=['HS256'])
|
||||||
|
exp = res_dict.get('exp')
|
||||||
|
exp_time = time.ctime(exp)
|
||||||
|
return exp_time
|
||||||
|
except Exception as e:
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def update_tp_querydict(querydict):
|
||||||
|
new_querydict = dict()
|
||||||
|
if 'record_time' in querydict:
|
||||||
|
new_querydict['record_time__icontains'] = ''.join(querydict.pop('record_time'))
|
||||||
|
if 'police_id' in querydict:
|
||||||
|
new_querydict['police_id__icontains'] = ''.join(querydict.pop('police_id'))
|
||||||
|
if 'event_type' in querydict:
|
||||||
|
new_querydict['event_type__icontains'] = ''.join(querydict.pop('event_type'))
|
||||||
|
if 'start_time' in querydict:
|
||||||
|
new_querydict['record_time__gte'] = ''.join(querydict.pop('start_time'))
|
||||||
|
if 'end_time' in querydict:
|
||||||
|
new_querydict['record_time__lte'] = datetime.datetime.strptime(''.join(querydict.pop('end_time')), "%Y-%m-%d") + datetime.timedelta(days=1)
|
||||||
|
if 'violation' in querydict:
|
||||||
|
if ''.join(querydict.get('violation')) == '1':
|
||||||
|
new_querydict['is_violation'] = True
|
||||||
|
elif ''.join(querydict.get('violation')) == '0':
|
||||||
|
new_querydict['is_violation'] = False
|
||||||
|
if 'violation_type' in querydict:
|
||||||
|
new_querydict['ai_analysis__icontains'] = ''.join(querydict.pop('violation_type'))
|
||||||
|
return new_querydict
|
||||||
|
|
||||||
|
|
||||||
|
def get_file_hash(filename):
|
||||||
|
if path.isfile(filename) is False:
|
||||||
|
return
|
||||||
|
|
||||||
|
# make a hash object
|
||||||
|
h_sha256 = hashlib.sha256()
|
||||||
|
|
||||||
|
# open file for reading in binary mode
|
||||||
|
with open(filename, "rb") as file:
|
||||||
|
# read file in chunks and update hash
|
||||||
|
chunk = 0
|
||||||
|
while chunk != b"":
|
||||||
|
chunk = file.read(1024)
|
||||||
|
h_sha256.update(chunk)
|
||||||
|
|
||||||
|
# return the hex digest
|
||||||
|
return h_sha256.hexdigest()
|
||||||
@ -0,0 +1,3 @@
|
|||||||
|
from django.contrib import admin
|
||||||
|
|
||||||
|
# Register your models here.
|
||||||
@ -0,0 +1,6 @@
|
|||||||
|
from django.apps import AppConfig
|
||||||
|
|
||||||
|
|
||||||
|
class DepartmentConfig(AppConfig):
|
||||||
|
default_auto_field = 'django.db.models.BigAutoField'
|
||||||
|
name = 'department'
|
||||||
@ -0,0 +1,27 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-09-22 09:47
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
initial = True
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='Department',
|
||||||
|
fields=[
|
||||||
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('name', models.CharField(max_length=128, verbose_name='部门名称')),
|
||||||
|
('parent_id', models.IntegerField(blank=True, null=True, verbose_name='上级部门id')),
|
||||||
|
('status', models.IntegerField(default=1, verbose_name='状态')),
|
||||||
|
('create_time', models.DateTimeField(auto_now_add=True, null=True, verbose_name='创建时间')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'db_table': 'tp_department',
|
||||||
|
},
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,14 @@
|
|||||||
|
from django.db import models
|
||||||
|
|
||||||
|
# Create your models here.
|
||||||
|
|
||||||
|
|
||||||
|
class Department(models.Model):
|
||||||
|
name = models.CharField(max_length=128, verbose_name='部门名称')
|
||||||
|
parent_id = models.IntegerField(verbose_name='上级部门id', null=True, blank=True)
|
||||||
|
status = models.IntegerField(verbose_name='状态', default=1)
|
||||||
|
create_time = models.DateTimeField(auto_now_add=True, verbose_name='创建时间', null=True, blank=True)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = 'tp_department'
|
||||||
|
|
||||||
@ -0,0 +1,9 @@
|
|||||||
|
from rest_framework import serializers
|
||||||
|
|
||||||
|
from .models import Department
|
||||||
|
|
||||||
|
|
||||||
|
class DepartmentSerializer(serializers.ModelSerializer):
|
||||||
|
class Meta:
|
||||||
|
model = Department
|
||||||
|
fields = "__all__"
|
||||||
@ -0,0 +1,3 @@
|
|||||||
|
from django.test import TestCase
|
||||||
|
|
||||||
|
# Create your tests here.
|
||||||
@ -0,0 +1,11 @@
|
|||||||
|
from django.urls import path
|
||||||
|
from .views import DepartmentView
|
||||||
|
|
||||||
|
urlpatterns = [
|
||||||
|
path('', DepartmentView.as_view({'get': 'list', 'post': 'create'})),
|
||||||
|
path('<pk>/', DepartmentView.as_view({'put': 'update', 'delete': 'destroy'})),
|
||||||
|
# path('query_depart_users/', DepartmentView.as_view({'post': 'query_department_users'})),
|
||||||
|
# path('son_departments/', DepartmentView.as_view({'post': 'create_son_department'})),
|
||||||
|
|
||||||
|
]
|
||||||
|
|
||||||
@ -0,0 +1,159 @@
|
|||||||
|
import logging
|
||||||
|
from django.db import transaction
|
||||||
|
from django.db.models import Q
|
||||||
|
from django.shortcuts import render
|
||||||
|
from rest_framework.response import Response
|
||||||
|
from rest_framework.views import APIView
|
||||||
|
from rest_framework import viewsets
|
||||||
|
from app.pagination import MyPageNumberPagination
|
||||||
|
from user.models import UserProfile
|
||||||
|
from .models import Department
|
||||||
|
from .serializers import DepartmentSerializer
|
||||||
|
# Create your views here.
|
||||||
|
|
||||||
|
logger = logging.getLogger('mylogger')
|
||||||
|
|
||||||
|
|
||||||
|
class DepartmentView(viewsets.GenericViewSet):
|
||||||
|
def list(self, request, *args, **kwargs):
|
||||||
|
data = dict()
|
||||||
|
try:
|
||||||
|
parent_departments = list(Department.objects.filter(parent_id__isnull=True).values())
|
||||||
|
for parent_department in parent_departments:
|
||||||
|
parent_id = parent_department.get('id')
|
||||||
|
son_departs = list(Department.objects.filter(parent_id=parent_id).values())
|
||||||
|
parent_department['children'] = son_departs
|
||||||
|
data['data'] = parent_departments
|
||||||
|
return Response(data)
|
||||||
|
except Exception as e:
|
||||||
|
return Response({'msg': f'查询失败, {e}'})
|
||||||
|
|
||||||
|
def query_department_users(self, request, *args, **kwargs):
|
||||||
|
res = dict()
|
||||||
|
data = request.data
|
||||||
|
department_id = data.get('department_id')
|
||||||
|
try:
|
||||||
|
users = list(UserProfile.objects.filter(department_id=department_id).values('id', 'username', 'phone_number', 'status', 'department_id'))
|
||||||
|
res['data'] = users
|
||||||
|
return Response(res)
|
||||||
|
except Exception as e:
|
||||||
|
return Response({'msg': f'查询失败，原因{e}'})
|
||||||
|
|
||||||
|
# def list_son_departments(self, request, *args, **kwargs):
|
||||||
|
# query_params = dict(request.query_params)
|
||||||
|
# parent_name = query_params.get('parent_name')
|
||||||
|
# if isinstance(parent_name, list):
|
||||||
|
# parent_name = ''.join(parent_name)
|
||||||
|
# logger.info(parent_name)
|
||||||
|
# parent_obj = Department.objects.filter(name=parent_name).first()
|
||||||
|
# if not parent_obj:
|
||||||
|
# return Response({'msg': '非法请求，该父级部门不存在'})
|
||||||
|
# parent_id = parent_obj.id
|
||||||
|
# queryset = Department.objects.filter(parent_id=parent_id).order_by('-id')
|
||||||
|
# paginator = MyPageNumberPagination()
|
||||||
|
# paginated_queryset = paginator.paginate_queryset(queryset, request)
|
||||||
|
# serializer = DepartmentSerializer(paginated_queryset, many=True)
|
||||||
|
#
|
||||||
|
# return paginator.get_paginated_response(serializer.data)
|
||||||
|
|
||||||
|
# def create(self, request, *args, **kwargs):
|
||||||
|
# data = request.data
|
||||||
|
# parent_depart_name = data.get('parent_department')
|
||||||
|
# if not parent_depart_name:
|
||||||
|
# return Response({"msg": "请输入父级部门名称"})
|
||||||
|
# parent_obj = Department.objects.filter(name=parent_depart_name).first()
|
||||||
|
# if parent_obj:
|
||||||
|
# return Response({'msg': '该部门已存在，请换一个'})
|
||||||
|
# try:
|
||||||
|
# with transaction.atomic():
|
||||||
|
# new_obj = Department.objects.create(name=parent_depart_name)
|
||||||
|
# return Response({'msg': '创建父级部门成功'})
|
||||||
|
# except Exception as e:
|
||||||
|
# return Response({'msg': f'创建失败，原因{e}'})
|
||||||
|
|
||||||
|
# def create(self, request, *args, **kwargs):
|
||||||
|
# data = request.data
|
||||||
|
# parent_name = data.get('parent_department_name')
|
||||||
|
# if not parent_name:
|
||||||
|
# return Response({'msg': '请输入父级部门名称'})
|
||||||
|
# depart_name = data.get('department_name')
|
||||||
|
# try:
|
||||||
|
# with transaction.atomic():
|
||||||
|
# if not depart_name:
|
||||||
|
# old_parent_obj = Department.objects.filter(name=parent_name).first()
|
||||||
|
# if old_parent_obj:
|
||||||
|
# return Response({'msg': '该父级部门名称已存在，请换一个'})
|
||||||
|
# parent_obj = Department.objects.create(name=parent_name)
|
||||||
|
# return Response({'msg': '创建父级部门成功'})
|
||||||
|
# else:
|
||||||
|
# old_parent_obj = Department.objects.filter(name=parent_name).first()
|
||||||
|
# if not old_parent_obj:
|
||||||
|
# old_parent_obj = Department.objects.create(name=parent_name)
|
||||||
|
# old_depart_obj = Department.objects.filter(name=depart_name, parent_id=old_parent_obj.id).first()
|
||||||
|
# if old_depart_obj:
|
||||||
|
# return Response({'msg': f'{parent_name}下已有{depart_name}啦，请换一个'})
|
||||||
|
# new_depart_obj = Department.objects.create(name=depart_name, parent_id=old_parent_obj.id)
|
||||||
|
# return Response({'msg': '创建子部门成功'})
|
||||||
|
# except Exception as e:
|
||||||
|
# return Response({'msg': f'创建失败，{e}'})
|
||||||
|
|
||||||
|
def create(self, request, *args, **kwargs):
|
||||||
|
data = request.data
|
||||||
|
department_name = data.get('department_name')
|
||||||
|
department_id = data.get('department_id')
|
||||||
|
if not department_id or department_id == '':
|
||||||
|
old_depart_obj = Department.objects.filter(name=department_name, parent_id__isnull=True).first()
|
||||||
|
if old_depart_obj:
|
||||||
|
return Response({'msg': f'该父级部门{department_name}已存在'})
|
||||||
|
depart_obj = Department.objects.create(name=department_name)
|
||||||
|
return Response({'msg': f'创建父级部门{department_name}成功', 'success': True})
|
||||||
|
else:
|
||||||
|
parent_obj = Department.objects.filter(id=department_id, parent_id__isnull=True).first()
|
||||||
|
if not parent_obj:
|
||||||
|
return Response({'msg': '非法请求'})
|
||||||
|
old_son_obj = Department.objects.filter(name=department_name, parent_id=department_id).first()
|
||||||
|
if old_son_obj:
|
||||||
|
return Response({'msg': f'该父级部门{parent_obj.name}下已有{old_son_obj.name}啦'})
|
||||||
|
son_obj = Department.objects.create(name=department_name, parent_id=department_id)
|
||||||
|
return Response({'msg': f'创建父级部门{parent_obj.name}下{son_obj.name}成功', 'success': True})
|
||||||
|
|
||||||
|
# def create_son_department(self, request, *args, **kwargs):
|
||||||
|
# data = request.data
|
||||||
|
# parent_name = data.get('parent_name')
|
||||||
|
# if not parent_name:
|
||||||
|
# return Response({'msg': '请输入父级部门名称以便创建子部门'})
|
||||||
|
# parent_obj = Department.objects.filter(name=parent_name).first()
|
||||||
|
# parent_id = parent_obj.id
|
||||||
|
# department_name = data.get('department_name')
|
||||||
|
# if not department_name:
|
||||||
|
# return Response({'msg': '请输入子部门名称'})
|
||||||
|
# depart_obj = Department.objects.filter(name=department_name, parent_id=parent_id).first()
|
||||||
|
# if depart_obj:
|
||||||
|
# return Response({'msg': f'{parent_name}下已有{department_name}，请换一个'})
|
||||||
|
# try:
|
||||||
|
# with transaction.atomic():
|
||||||
|
# Department.objects.create(name=department_name, parent_id=parent_id)
|
||||||
|
# return Response({'msg': '创建成功'})
|
||||||
|
# except Exception as e:
|
||||||
|
# return Response({'msg': f'创建失败，原因{e}'})
|
||||||
|
|
||||||
|
def update(self, request, pk, *args, **kwargs):
|
||||||
|
data = request.data
|
||||||
|
try:
|
||||||
|
with transaction.atomic():
|
||||||
|
depart_obj = Department.objects.filter(id=pk).update(**data)
|
||||||
|
return Response({'msg': '更新成功', 'success': True})
|
||||||
|
except Department.DoesNotExist:
|
||||||
|
return Response({'msg': '该部门不存在'})
|
||||||
|
except Exception as e:
|
||||||
|
return Response({'msg': f'更新失败,原因{e}'})
|
||||||
|
|
||||||
|
def destroy(self, request, pk, *args, **kwargs):
|
||||||
|
try:
|
||||||
|
with transaction.atomic():
|
||||||
|
depart_obj = Department.objects.filter(id=pk).delete()
|
||||||
|
return Response({'msg': '删除成功', 'success': True})
|
||||||
|
except Department.DoesNotExist:
|
||||||
|
return Response({'msg': '该部门不存在'})
|
||||||
|
except Exception as e:
|
||||||
|
return Response({'msg': f'删除失败，原因{e}'})
|
||||||
@ -0,0 +1,33 @@
|
|||||||
|
# 输出查询结果])
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
|
from elasticsearch import Elasticsearch
|
||||||
|
|
||||||
|
# 连接到 Elasticsearch
|
||||||
|
client = Elasticsearch("http://localhost:9209", basic_auth=("elastic", "changeme"))
|
||||||
|
|
||||||
|
start_date = datetime(2024, 1, 1) # 开始时间
|
||||||
|
end_date = datetime(2024, 12, 31) # 结束时间
|
||||||
|
|
||||||
|
# 构建查询体
|
||||||
|
query = {
|
||||||
|
"query": {
|
||||||
|
"range": {
|
||||||
|
"@timestamp": {
|
||||||
|
"gte": start_date.strftime("%Y-%m-%dT%H:%M:%S"),
|
||||||
|
"lte": end_date.strftime("%Y-%m-%dT%H:%M:%S")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"size": 10000
|
||||||
|
}
|
||||||
|
|
||||||
|
# 执行查询操作
|
||||||
|
response = client.search(index="tp-log", body=query)
|
||||||
|
|
||||||
|
# 打印结果
|
||||||
|
print("Found {} documents:".format(response['hits']['total']['value']))
|
||||||
|
result = client.delete_by_query(index="tp-log", body=query)
|
||||||
|
|
||||||
|
# 打印删除结果
|
||||||
|
print("Deleted:", result)
|
||||||
@ -0,0 +1,5 @@
|
|||||||
|
from django.contrib import admin
|
||||||
|
|
||||||
|
# Register your models here.
|
||||||
|
from event.models import Event
|
||||||
|
admin.site.register(Event)
|
||||||
@ -0,0 +1,6 @@
|
|||||||
|
from django.apps import AppConfig
|
||||||
|
|
||||||
|
|
||||||
|
class EventConfig(AppConfig):
|
||||||
|
default_auto_field = 'django.db.models.BigAutoField'
|
||||||
|
name = 'event'
|
||||||
@ -0,0 +1,14 @@
|
|||||||
|
import django_filters
|
||||||
|
from django_filters.rest_framework import FilterSet
|
||||||
|
|
||||||
|
from event.models import Event
|
||||||
|
|
||||||
|
|
||||||
|
class EventFilter(FilterSet):
|
||||||
|
event_name = django_filters.CharFilter(field_name='event_name', lookup_expr='icontains')
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
# 指定模型
|
||||||
|
models = Event
|
||||||
|
# 指定需要模糊查询的字段
|
||||||
|
fields = ("event_name",)
|
||||||
@ -0,0 +1,27 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-09-22 09:47
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
initial = True
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='Event',
|
||||||
|
fields=[
|
||||||
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('event_name', models.CharField(default='', max_length=256, verbose_name='事件名')),
|
||||||
|
('create_time', models.DateTimeField(auto_now_add=True, verbose_name='创建时间')),
|
||||||
|
('update_time', models.DateTimeField(auto_now=True, verbose_name='更新时间')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'db_table': 'tp_event',
|
||||||
|
'ordering': ['-id'],
|
||||||
|
},
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,13 @@
|
|||||||
|
from django.db import models
|
||||||
|
|
||||||
|
# Create your models here.
|
||||||
|
|
||||||
|
|
||||||
|
class Event(models.Model):
|
||||||
|
event_name = models.CharField(max_length=256, default='', verbose_name='事件名')
|
||||||
|
create_time = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
|
||||||
|
update_time = models.DateTimeField(auto_now=True, verbose_name='更新时间')
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = 'tp_event'
|
||||||
|
ordering = ['-id']
|
||||||
@ -0,0 +1,8 @@
|
|||||||
|
from rest_framework import serializers
|
||||||
|
from event.models import Event
|
||||||
|
|
||||||
|
|
||||||
|
class EventSerializer(serializers.ModelSerializer):
|
||||||
|
class Meta:
|
||||||
|
model = Event
|
||||||
|
fields = "__all__"
|
||||||
@ -0,0 +1,3 @@
|
|||||||
|
from django.test import TestCase
|
||||||
|
|
||||||
|
# Create your tests here.
|
||||||
@ -0,0 +1,10 @@
|
|||||||
|
from django.urls import path, include
|
||||||
|
from rest_framework.routers import DefaultRouter
|
||||||
|
from event import views
|
||||||
|
|
||||||
|
router = DefaultRouter()
|
||||||
|
router.register('', views.EventViewSet)
|
||||||
|
|
||||||
|
urlpatterns = [
|
||||||
|
path('', include(router.urls)),
|
||||||
|
]
|
||||||
@ -0,0 +1,23 @@
|
|||||||
|
from django.shortcuts import render
|
||||||
|
from rest_framework import viewsets
|
||||||
|
|
||||||
|
from app.pagination import MyPageNumberPagination
|
||||||
|
from event.filters import EventFilter
|
||||||
|
from event.models import Event
|
||||||
|
from event.serializers import EventSerializer
|
||||||
|
|
||||||
|
|
||||||
|
# Create your views here.
|
||||||
|
|
||||||
|
class EventViewSet(viewsets.ModelViewSet):
|
||||||
|
# 查询集
|
||||||
|
queryset = Event.objects.all().order_by('-id')
|
||||||
|
# 序列化器
|
||||||
|
serializer_class = EventSerializer
|
||||||
|
# 分页器
|
||||||
|
pagination_class = MyPageNumberPagination
|
||||||
|
# 过滤器
|
||||||
|
filterset_class = EventFilter
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,22 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
"""Django's command-line utility for administrative tasks."""
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
"""Run administrative tasks."""
|
||||||
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'TP_API.settings')
|
||||||
|
try:
|
||||||
|
from django.core.management import execute_from_command_line
|
||||||
|
except ImportError as exc:
|
||||||
|
raise ImportError(
|
||||||
|
"Couldn't import Django. Are you sure it's installed and "
|
||||||
|
"available on your PYTHONPATH environment variable? Did you "
|
||||||
|
"forget to activate a virtual environment?"
|
||||||
|
) from exc
|
||||||
|
execute_from_command_line(sys.argv)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
||||||
@ -0,0 +1,6 @@
|
|||||||
|
[global]
|
||||||
|
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
|
||||||
|
[install]
|
||||||
|
trusted-host = https://pypi.tuna.tsinghua.edu.cn
|
||||||
|
[list]
|
||||||
|
format=columns
|
||||||
Binary file not shown.
@ -0,0 +1,9 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
echo "Appling database migrations..."
|
||||||
|
sleep 5
|
||||||
|
python manage.py makemigrations && python manage.py migrate && python manage.py runserver 0.0.0.0:8000&&
|
||||||
|
echo 'TP项目启动完成'
|
||||||
|
tail -f /dev/null
|
||||||
|
|
||||||
|
exec "$@"
|
||||||
Binary file not shown.
Binary file not shown.
|
After Width: | Height: | Size: 11 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 88 KiB |
@ -0,0 +1,3 @@
|
|||||||
|
from django.contrib import admin
|
||||||
|
|
||||||
|
# Register your models here.
|
||||||
@ -0,0 +1,6 @@
|
|||||||
|
from django.apps import AppConfig
|
||||||
|
|
||||||
|
|
||||||
|
class UserConfig(AppConfig):
|
||||||
|
default_auto_field = 'django.db.models.BigAutoField'
|
||||||
|
name = 'user'
|
||||||
@ -0,0 +1,51 @@
|
|||||||
|
# Generated by Django 3.2.19 on 2023-09-22 09:47
|
||||||
|
|
||||||
|
import django.contrib.auth.models
|
||||||
|
import django.contrib.auth.validators
|
||||||
|
from django.db import migrations, models
|
||||||
|
import django.db.models.deletion
|
||||||
|
import django.utils.timezone
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
initial = True
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('auth', '0012_alter_user_first_name_max_length'),
|
||||||
|
('department', '0001_initial'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='UserProfile',
|
||||||
|
fields=[
|
||||||
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('password', models.CharField(max_length=128, verbose_name='password')),
|
||||||
|
('last_login', models.DateTimeField(blank=True, null=True, verbose_name='last login')),
|
||||||
|
('is_superuser', models.BooleanField(default=False, help_text='Designates that this user has all permissions without explicitly assigning them.', verbose_name='superuser status')),
|
||||||
|
('username', models.CharField(error_messages={'unique': 'A user with that username already exists.'}, help_text='Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only.', max_length=150, unique=True, validators=[django.contrib.auth.validators.UnicodeUsernameValidator()], verbose_name='username')),
|
||||||
|
('first_name', models.CharField(blank=True, max_length=150, verbose_name='first name')),
|
||||||
|
('last_name', models.CharField(blank=True, max_length=150, verbose_name='last name')),
|
||||||
|
('email', models.EmailField(blank=True, max_length=254, verbose_name='email address')),
|
||||||
|
('is_staff', models.BooleanField(default=False, help_text='Designates whether the user can log into this admin site.', verbose_name='staff status')),
|
||||||
|
('is_active', models.BooleanField(default=True, help_text='Designates whether this user should be treated as active. Unselect this instead of deleting accounts.', verbose_name='active')),
|
||||||
|
('date_joined', models.DateTimeField(default=django.utils.timezone.now, verbose_name='date joined')),
|
||||||
|
('organization', models.CharField(blank=True, max_length=256, null=True, verbose_name='组织')),
|
||||||
|
('gender', models.IntegerField(choices=[(0, '女'), (1, '男')], default=1, verbose_name='性别')),
|
||||||
|
('phone_number', models.CharField(blank=True, max_length=12, null=True, verbose_name='手机号')),
|
||||||
|
('head_sculpture', models.CharField(blank=True, max_length=512, null=True, verbose_name='头像')),
|
||||||
|
('status', models.IntegerField(default=1, verbose_name='状态')),
|
||||||
|
('recorder_number', models.CharField(blank=True, max_length=256, null=True, verbose_name='执法记录仪编号')),
|
||||||
|
('department', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='department.department')),
|
||||||
|
('groups', models.ManyToManyField(blank=True, help_text='The groups this user belongs to. A user will get all permissions granted to each of their groups.', related_name='user_set', related_query_name='user', to='auth.Group', verbose_name='groups')),
|
||||||
|
('user_permissions', models.ManyToManyField(blank=True, help_text='Specific permissions for this user.', related_name='user_set', related_query_name='user', to='auth.Permission', verbose_name='user permissions')),
|
||||||
|
],
|
||||||
|
options={
|
||||||
|
'db_table': 'tp_user',
|
||||||
|
},
|
||||||
|
managers=[
|
||||||
|
('objects', django.contrib.auth.models.UserManager()),
|
||||||
|
],
|
||||||
|
),
|
||||||
|
]
|
||||||
@ -0,0 +1,28 @@
|
|||||||
|
from django.contrib.auth.models import AbstractUser
|
||||||
|
from django.db import models
|
||||||
|
from department.models import Department
|
||||||
|
|
||||||
|
# Create your models here.
|
||||||
|
|
||||||
|
|
||||||
|
# class Department(models.Model):
|
||||||
|
# department_name = models.CharField(max_length=128, verbose_name='部门名称')
|
||||||
|
# # parent_id = models.IntegerField(verbose_name='上级部门id', null=True, blank=True)
|
||||||
|
# # status = models.IntegerField(verbose_name='状态', default=1)
|
||||||
|
#
|
||||||
|
# class Meta:
|
||||||
|
# db_table = 'user_department'
|
||||||
|
|
||||||
|
|
||||||
|
class UserProfile(AbstractUser):
|
||||||
|
GENDER = ((0, '女'), (1, '男'))
|
||||||
|
organization = models.CharField(max_length=256, verbose_name='组织', null=True, blank=True)
|
||||||
|
gender = models.IntegerField(choices=GENDER, default=1, verbose_name='性别')
|
||||||
|
phone_number = models.CharField(max_length=12, verbose_name='手机号', null=True, blank=True)
|
||||||
|
head_sculpture = models.CharField(max_length=512, verbose_name='头像', blank=True, null=True)
|
||||||
|
status = models.IntegerField(verbose_name='状态', default=1)
|
||||||
|
recorder_number = models.CharField(max_length=256, verbose_name='执法记录仪编号', null=True, blank=True)
|
||||||
|
department = models.ForeignKey(Department, on_delete=models.CASCADE, null=True)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
db_table = 'tp_user'
|
||||||
@ -0,0 +1,12 @@
|
|||||||
|
from rest_framework import serializers
|
||||||
|
from .models import UserProfile
|
||||||
|
|
||||||
|
|
||||||
|
class UserSerializer(serializers.ModelSerializer):
|
||||||
|
class Meta:
|
||||||
|
model = UserProfile
|
||||||
|
# fields = "__all__"
|
||||||
|
|
||||||
|
exclude = ('password',)
|
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,278 @@
|
|||||||
|
from django.test import TestCase
|
||||||
|
|
||||||
|
# Create your tests here.
|
||||||
|
|
||||||
|
l1 = [
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-14 15:43:25",
|
||||||
|
"police_id": "8888888",
|
||||||
|
"event_type": "卡车",
|
||||||
|
"is_violation": False,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/B2_006.png",
|
||||||
|
"relative_time": 6.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/B2.mp4",
|
||||||
|
"car_number": "赣A·98980",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 09:17:45",
|
||||||
|
"update_time": "2023-06-27 02:36:51",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-14 15:43:24",
|
||||||
|
"police_id": "8888888",
|
||||||
|
"event_type": "卡车",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/B2_004.png",
|
||||||
|
"relative_time": 4.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/B2.mp4",
|
||||||
|
"car_number": "赣A·98980",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 09:16:41",
|
||||||
|
"update_time": "2023-06-27 02:36:51",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2023-05-14 17:27:40",
|
||||||
|
"police_id": "",
|
||||||
|
"event_type": "推搡",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/shoving_005_1.png",
|
||||||
|
"relative_time": 9.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/shoving.mp4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 07:13:24",
|
||||||
|
"update_time": "2023-06-27 02:36:51",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2023-05-14 17:27:40",
|
||||||
|
"police_id": "",
|
||||||
|
"event_type": "推搡",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/shoving_009_1.png",
|
||||||
|
"relative_time": 9.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/shoving.mp4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 07:13:19",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2023-05-14 17:27:40",
|
||||||
|
"police_id": "",
|
||||||
|
"event_type": "推搡",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/shoving_009.png",
|
||||||
|
"relative_time": 9.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/shoving.mp4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 07:12:34",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2023-05-14 17:27:40",
|
||||||
|
"police_id": "",
|
||||||
|
"event_type": "推搡",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/shoving_005.png",
|
||||||
|
"relative_time": 5.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/shoving.mp4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 07:10:03",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-14 17:27:40",
|
||||||
|
"police_id": "8888888",
|
||||||
|
"event_type": "未戴头盔",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/G2_018_1.png",
|
||||||
|
"relative_time": 18.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/G2.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:53:50",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-14 17:27:40",
|
||||||
|
"police_id": "",
|
||||||
|
"event_type": "未戴头盔",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/G2_018.png",
|
||||||
|
"relative_time": 18.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/G2.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:53:26",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-15 07:42:48",
|
||||||
|
"police_id": "0890151",
|
||||||
|
"event_type": "未戴头盔以及载人",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/G1_024_1.png",
|
||||||
|
"relative_time": 24.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/G1.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:52:12",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-15 07:42:48",
|
||||||
|
"police_id": "0890151",
|
||||||
|
"event_type": "未戴头盔以及载人",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/G1_024.png",
|
||||||
|
"relative_time": 24.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/G1.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:51:33",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-15 07:42:46",
|
||||||
|
"police_id": "0890151",
|
||||||
|
"event_type": "未戴头盔以及载人",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/G1_022.png",
|
||||||
|
"relative_time": 22.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/G1.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:50:53",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-15 07:42:28",
|
||||||
|
"police_id": "0890151",
|
||||||
|
"event_type": "未戴头盔以及载人",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/G1_004.png",
|
||||||
|
"relative_time": 4.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/G1.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:49:36",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-15 07:44:05",
|
||||||
|
"police_id": "0890151",
|
||||||
|
"event_type": "未戴头盔",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/G1_141_1.png",
|
||||||
|
"relative_time": 101.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/G1.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:48:34",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-15 07:44:05",
|
||||||
|
"police_id": "0890151",
|
||||||
|
"event_type": "未戴头盔",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/G1_141.jpg",
|
||||||
|
"relative_time": 141.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/G1.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:46:45",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-15 07:42:26",
|
||||||
|
"police_id": "0890151",
|
||||||
|
"event_type": "未戴头盔以及载人",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/G1_002.png",
|
||||||
|
"relative_time": 2.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/G1.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:45:39",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-15 10:36:13",
|
||||||
|
"police_id": "6666666",
|
||||||
|
"event_type": "未戴头盔",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/B1_056.png",
|
||||||
|
"relative_time": 56.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/B1.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:44:04",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"video_hash": "vbhdrbvcw",
|
||||||
|
"record_time": "2022-07-15 10:36:13",
|
||||||
|
"police_id": "6666666",
|
||||||
|
"event_type": "未戴头盔",
|
||||||
|
"is_violation": True,
|
||||||
|
"small_image": "http://192.168.10.28:8000/media/images/056.png",
|
||||||
|
"relative_time": 56.0,
|
||||||
|
"video_dir": "http://192.168.10.28:8000/media/videos/B1.MP4",
|
||||||
|
"car_number": "",
|
||||||
|
"ai_analysis": "",
|
||||||
|
"add_time": "2023-06-12 06:18:40",
|
||||||
|
"update_time": "2023-06-27 02:36:52",
|
||||||
|
"is_display": True
|
||||||
|
}
|
||||||
|
]
|
||||||
|
|
||||||
|
l2 = []
|
||||||
|
|
||||||
|
from app.models import TP
|
||||||
|
|
||||||
|
|
||||||
|
def import_data(data):
|
||||||
|
try:
|
||||||
|
for item in data:
|
||||||
|
TP.objects.create(**item)
|
||||||
|
return
|
||||||
|
except Exception as e:
|
||||||
|
print(e)
|
||||||
|
return
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import_data(l1)
|
||||||
@ -0,0 +1,13 @@
|
|||||||
|
from django.urls import path
|
||||||
|
from .views import UserViewSet
|
||||||
|
|
||||||
|
urlpatterns = [
|
||||||
|
path('', UserViewSet.as_view({"get": "query_users", "post": "create"})),
|
||||||
|
path('login/', UserViewSet.as_view({"post": "tp_login"})),
|
||||||
|
path('<pk>/', UserViewSet.as_view({"put": "update_user", "delete": "delete_user"})),
|
||||||
|
# path('set_user_role', UserViewSet.as_view({"post": "set_user_role"})),
|
||||||
|
# path('role/', RoleView.as_view()),
|
||||||
|
# path('role/<pk>/', RoleView.as_view())
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,13 @@
|
|||||||
|
def update_user_querydict(querydict):
|
||||||
|
new_querydict = dict()
|
||||||
|
if 'username' in querydict:
|
||||||
|
new_querydict['username__icontains'] = ''.join(querydict.pop('username'))
|
||||||
|
if 'phone_number' in querydict and querydict.get('phone_number') != ['']:
|
||||||
|
new_querydict['phone_number__icontains'] = ''.join(querydict.pop('phone_number'))
|
||||||
|
if 'status' in querydict and querydict.get('status') != ['']:
|
||||||
|
new_querydict['status'] = ''.join(querydict.pop('status'))
|
||||||
|
return new_querydict
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,242 @@
|
|||||||
|
import json
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from django.contrib.auth import authenticate
|
||||||
|
from django.contrib.auth.hashers import make_password
|
||||||
|
from django.db import transaction
|
||||||
|
from django.shortcuts import render
|
||||||
|
|
||||||
|
# Create your views here.
|
||||||
|
from rest_framework import viewsets
|
||||||
|
from rest_framework.response import Response
|
||||||
|
from rest_framework.views import APIView
|
||||||
|
from django.contrib.auth.models import Group
|
||||||
|
from app.pagination import MyPageNumberPagination
|
||||||
|
from app.utils import generate_token, decode_token_exp_time
|
||||||
|
from department.models import Department
|
||||||
|
from .models import UserProfile
|
||||||
|
from .serializers import UserSerializer
|
||||||
|
from .utils import update_user_querydict
|
||||||
|
|
||||||
|
logger = logging.getLogger('mylogger')
|
||||||
|
|
||||||
|
|
||||||
|
class UserViewSet(viewsets.GenericViewSet):
|
||||||
|
def tp_login(self, request, *args, **kwargs):
|
||||||
|
"""登录 POST"""
|
||||||
|
data = request.data
|
||||||
|
username = data.get('username')
|
||||||
|
password = data.get('password')
|
||||||
|
login_user = authenticate(username=username, password=password)
|
||||||
|
if login_user and login_user.is_active and login_user.status == 1:
|
||||||
|
# 生成token
|
||||||
|
token = generate_token(login_user)
|
||||||
|
response = {
|
||||||
|
'success': True,
|
||||||
|
'msg': '登录成功',
|
||||||
|
'data': {
|
||||||
|
'username': login_user.username,
|
||||||
|
'roles': ['admin'] if login_user.is_superuser else ['common'],
|
||||||
|
'accessToken': token,
|
||||||
|
'expires': decode_token_exp_time(token)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return Response(response)
|
||||||
|
else:
|
||||||
|
response = {
|
||||||
|
'success': False,
|
||||||
|
'msg': '登录失败',
|
||||||
|
'data': {
|
||||||
|
'username': None,
|
||||||
|
'roles': [],
|
||||||
|
'accessToken': None,
|
||||||
|
'expires': None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return Response(response)
|
||||||
|
|
||||||
|
def create(self, request, *args, **kwargs):
|
||||||
|
"""注册 POST 用户名 手机号 状态 性别 部门"""
|
||||||
|
data = request.data
|
||||||
|
username = data.get('username')
|
||||||
|
if not username:
|
||||||
|
return Response({'msg': '用户名不能为空'})
|
||||||
|
phone_number = data.get('phone_number')
|
||||||
|
gender = data.get('gender')
|
||||||
|
password = data.get('password', '#Yaxin0504')
|
||||||
|
# role = data.get('role')
|
||||||
|
# role_obj = Group.objects.filter(name=role).first()
|
||||||
|
department_id = data.get('department_id')
|
||||||
|
department_obj = Department.objects.filter(id=department_id).first()
|
||||||
|
status = data.get('status', 1)
|
||||||
|
try:
|
||||||
|
if UserProfile.objects.filter(username=username).first():
|
||||||
|
return Response({'msg': '该用户名已存在，请换一个'})
|
||||||
|
user = UserProfile.objects.create_user(username=username, password=password, department=department_obj, status=status, gender=gender, phone_number=phone_number)
|
||||||
|
# user.groups.add(role_obj)
|
||||||
|
# user.save()
|
||||||
|
return Response({'msg': '添加成功', 'success': True})
|
||||||
|
except Exception as e:
|
||||||
|
response = {
|
||||||
|
"msg": f'添加失败, 原因:{e}'
|
||||||
|
}
|
||||||
|
return Response(response)
|
||||||
|
|
||||||
|
def query_users(self, request, *args, **kwargs):
|
||||||
|
"""查询所有用户，默认全部，可根据部门查询"""
|
||||||
|
query_params = dict(request.query_params)
|
||||||
|
logger.info(query_params)
|
||||||
|
querydict = update_user_querydict(query_params)
|
||||||
|
logger.info(querydict)
|
||||||
|
if query_params.get('department_id') and query_params.get('department_id') != ['']:
|
||||||
|
department_id = ''.join(query_params.get('department_id'))
|
||||||
|
depart_obj = Department.objects.filter(id=department_id).first()
|
||||||
|
if not depart_obj.parent_id:
|
||||||
|
# 父级部门
|
||||||
|
dp_ids = list(Department.objects.filter(parent_id=depart_obj.id).values_list('id', flat=True).distinct())
|
||||||
|
else:
|
||||||
|
dp_ids = [depart_obj.id]
|
||||||
|
queryset = UserProfile.objects.filter(department_id__in=dp_ids, **querydict).order_by('-id')
|
||||||
|
else:
|
||||||
|
queryset = UserProfile.objects.filter(**querydict).order_by('-id')
|
||||||
|
paginator = MyPageNumberPagination()
|
||||||
|
paginated_queryset = paginator.paginate_queryset(queryset, request)
|
||||||
|
serializer = UserSerializer(paginated_queryset, many=True)
|
||||||
|
paginator_response = paginator.get_paginated_response(serializer.data)
|
||||||
|
response_data = {
|
||||||
|
'count': paginator_response.data['count'],
|
||||||
|
'next': paginator_response.data['next'],
|
||||||
|
'previous': paginator_response.data['previous'],
|
||||||
|
'results': list(paginator_response.data['results']),
|
||||||
|
}
|
||||||
|
results = response_data.get('results')
|
||||||
|
res_list = json.loads(json.dumps(results))
|
||||||
|
for res_dict in res_list:
|
||||||
|
department_obj = Department.objects.filter(id=res_dict.get('department')).first()
|
||||||
|
if department_obj:
|
||||||
|
res_dict['department_name'] = department_obj.name
|
||||||
|
res_dict['department_id'] = res_dict.pop('department')
|
||||||
|
else:
|
||||||
|
res_dict['department_name'] = '暂无部门'
|
||||||
|
res_dict['department_id'] = '暂无'
|
||||||
|
response_data['results'] = res_list
|
||||||
|
return Response(response_data)
|
||||||
|
|
||||||
|
# def set_user_role(self, request, *args, **kwargs):
|
||||||
|
# try:
|
||||||
|
# user_id = request.data.get('user_id')
|
||||||
|
# user = UserProfile.objects.get(id=user_id)
|
||||||
|
# group = Group.objects.get(name=request.data.get('role_name'))
|
||||||
|
# is_member = group.user_set.filter(id=user.id).exists()
|
||||||
|
# if is_member:
|
||||||
|
# return Response({'msg': f'用户{user.username}已经是{group.name}啦'})
|
||||||
|
# user.groups.add(group)
|
||||||
|
# user.save()
|
||||||
|
# return Response({'msg': '修改用户角色成功'})
|
||||||
|
# except UserProfile.DoesNotExist:
|
||||||
|
# return Response({'msg': '该用户不存在'})
|
||||||
|
# except Group.DoesNotExist:
|
||||||
|
# return Response({'msg': '该角色不存在'})
|
||||||
|
# except Exception as e:
|
||||||
|
# return Response({'msg': f'更改用户角色失败: {e}'})
|
||||||
|
|
||||||
|
def update_user(self, request, pk, *args, **kwargs):
|
||||||
|
data = request.data
|
||||||
|
if 'username' in data:
|
||||||
|
user_obj = UserProfile.objects.exclude(id=pk).filter(username=data.get('username')).first()
|
||||||
|
if user_obj:
|
||||||
|
return Response({'msg': '更新失败，该用户名已存在，请换一个'})
|
||||||
|
# user_id = data.pop('user_id', None)
|
||||||
|
if 'password' in data:
|
||||||
|
# hashed_pwd = make_password(data.get('password'))
|
||||||
|
data['password'] = make_password(data.pop('password'))
|
||||||
|
try:
|
||||||
|
with transaction.atomic():
|
||||||
|
user = UserProfile.objects.filter(id=pk).update(**data)
|
||||||
|
return Response({'msg': '更新成功', 'success': True})
|
||||||
|
except UserProfile.DoesNotExist:
|
||||||
|
return Response({'msg': '该用户不存在'})
|
||||||
|
except Exception as e:
|
||||||
|
return Response({'msg': f'更新失败,原因{e}'})
|
||||||
|
|
||||||
|
def delete_user(self, request, pk, *args, **kwargs):
|
||||||
|
try:
|
||||||
|
with transaction.atomic():
|
||||||
|
user = UserProfile.objects.filter(id=pk).delete()
|
||||||
|
return Response({'msg': '删除成功', 'success': True})
|
||||||
|
except UserProfile.DoesNotExist:
|
||||||
|
return Response({'msg': '该用户不存在'})
|
||||||
|
except Exception as e:
|
||||||
|
return Response({'msg': f'删除失败，原因{e}'})
|
||||||
|
|
||||||
|
|
||||||
|
# class RoleView(APIView):
|
||||||
|
# def get(self, request):
|
||||||
|
# queryset = Group.objects.all().order_by('-id')
|
||||||
|
# paginator = MyPageNumberPagination()
|
||||||
|
# paginated_queryset = paginator.paginate_queryset(queryset, request)
|
||||||
|
#
|
||||||
|
# results = []
|
||||||
|
# for group in paginated_queryset:
|
||||||
|
# results.append({
|
||||||
|
# 'id': group.id,
|
||||||
|
# 'name': group.name,
|
||||||
|
# })
|
||||||
|
#
|
||||||
|
# response = {
|
||||||
|
# 'count': paginator.page.paginator.count,
|
||||||
|
# 'next': paginator.get_next_link(),
|
||||||
|
# 'previous': paginator.get_previous_link(),
|
||||||
|
# 'results': results
|
||||||
|
# }
|
||||||
|
#
|
||||||
|
# return Response(response)
|
||||||
|
#
|
||||||
|
# def post(self, request, *args, **kwargs):
|
||||||
|
# """添加角色"""
|
||||||
|
# data = request.data
|
||||||
|
# group_name = data.get('role_name')
|
||||||
|
# try:
|
||||||
|
# old_group = Group.objects.filter(name=group_name).first()
|
||||||
|
# if old_group:
|
||||||
|
# return Response({"msg": "该角色名已存在"})
|
||||||
|
# with transaction.atomic():
|
||||||
|
# group = Group.objects.create(name=group_name)
|
||||||
|
# group.save()
|
||||||
|
# return Response({"msg": "创建成功"})
|
||||||
|
# except Exception as e:
|
||||||
|
# return Response({"msg": f"创建失败:{e}"})
|
||||||
|
#
|
||||||
|
# #
|
||||||
|
# def put(self, request, pk):
|
||||||
|
# group_name = request.data.get('role_name')
|
||||||
|
# try:
|
||||||
|
# with transaction.atomic(): # 开启事务
|
||||||
|
# group = Group.objects.get(id=pk)
|
||||||
|
# group.name = group_name
|
||||||
|
# group.save()
|
||||||
|
# return Response({'success': True, 'msg': 'Group updated successfully'})
|
||||||
|
# except Group.DoesNotExist:
|
||||||
|
# return Response({'success': False, 'msg': 'Group not found'})
|
||||||
|
# except Exception as e:
|
||||||
|
# return Response({'success': False, 'msg': str(e)})
|
||||||
|
#
|
||||||
|
# def delete(self, request, pk):
|
||||||
|
# try:
|
||||||
|
# with transaction.atomic(): # 开启事务
|
||||||
|
# group = Group.objects.get(id=pk)
|
||||||
|
# group.delete()
|
||||||
|
# return Response({'success': True, 'msg': 'Group deleted successfully'})
|
||||||
|
# except Group.DoesNotExist:
|
||||||
|
# return Response({'success': False, 'msg': 'Group not found'})
|
||||||
|
# except Exception as e:
|
||||||
|
# return Response({'success': False, 'msg': str(e)})
|
||||||
|
#
|
||||||
|
# def modify_permission(self, request, *args, **kwargs):
|
||||||
|
# pass
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,192 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
# The MIT License (MIT)
|
||||||
|
#
|
||||||
|
# Copyright (c) 2017 Eficode Oy
|
||||||
|
#
|
||||||
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
# of this software and associated documentation files (the "Software"), to deal
|
||||||
|
# in the Software without restriction, including without limitation the rights
|
||||||
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
# copies of the Software, and to permit persons to whom the Software is
|
||||||
|
# furnished to do so, subject to the following conditions:
|
||||||
|
#
|
||||||
|
# The above copyright notice and this permission notice shall be included in all
|
||||||
|
# copies or substantial portions of the Software.
|
||||||
|
#
|
||||||
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
# SOFTWARE.
|
||||||
|
|
||||||
|
VERSION="2.2.4"
|
||||||
|
|
||||||
|
set -- "$@" -- "$TIMEOUT" "$QUIET" "$PROTOCOL" "$HOST" "$PORT" "$result"
|
||||||
|
TIMEOUT=15
|
||||||
|
QUIET=0
|
||||||
|
# The protocol to make the request with, either "tcp" or "http"
|
||||||
|
PROTOCOL="tcp"
|
||||||
|
|
||||||
|
echoerr() {
|
||||||
|
if [ "$QUIET" -ne 1 ]; then printf "%s\n" "$*" 1>&2; fi
|
||||||
|
}
|
||||||
|
|
||||||
|
usage() {
|
||||||
|
exitcode="$1"
|
||||||
|
cat << USAGE >&2
|
||||||
|
Usage:
|
||||||
|
$0 host:port|url [-t timeout] [-- command args]
|
||||||
|
-q | --quiet Do not output any status messages
|
||||||
|
-t TIMEOUT | --timeout=timeout Timeout in seconds, zero for no timeout
|
||||||
|
Defaults to 15 seconds
|
||||||
|
-v | --version Show the version of this tool
|
||||||
|
-- COMMAND ARGS Execute command with args after the test finishes
|
||||||
|
USAGE
|
||||||
|
exit "$exitcode"
|
||||||
|
}
|
||||||
|
|
||||||
|
wait_for() {
|
||||||
|
case "$PROTOCOL" in
|
||||||
|
tcp)
|
||||||
|
if ! command -v nc >/dev/null; then
|
||||||
|
echoerr 'nc command is missing!'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
http)
|
||||||
|
if ! command -v wget >/dev/null; then
|
||||||
|
echoerr 'wget command is missing!'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
TIMEOUT_END=$(($(date +%s) + TIMEOUT))
|
||||||
|
|
||||||
|
while :; do
|
||||||
|
case "$PROTOCOL" in
|
||||||
|
tcp)
|
||||||
|
nc -w 1 -z "$HOST" "$PORT" > /dev/null 2>&1
|
||||||
|
;;
|
||||||
|
http)
|
||||||
|
wget --timeout=1 --tries=1 -q "$HOST" -O /dev/null > /dev/null 2>&1
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echoerr "Unknown protocol '$PROTOCOL'"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
result=$?
|
||||||
|
|
||||||
|
if [ $result -eq 0 ] ; then
|
||||||
|
if [ $# -gt 7 ] ; then
|
||||||
|
for result in $(seq $(($# - 7))); do
|
||||||
|
result=$1
|
||||||
|
shift
|
||||||
|
set -- "$@" "$result"
|
||||||
|
done
|
||||||
|
|
||||||
|
TIMEOUT=$2 QUIET=$3 PROTOCOL=$4 HOST=$5 PORT=$6 result=$7
|
||||||
|
shift 7
|
||||||
|
exec "$@"
|
||||||
|
fi
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ $TIMEOUT -ne 0 -a $(date +%s) -ge $TIMEOUT_END ]; then
|
||||||
|
echo "Operation timed out" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
sleep 1
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
while :; do
|
||||||
|
case "$1" in
|
||||||
|
http://*|https://*)
|
||||||
|
HOST="$1"
|
||||||
|
PROTOCOL="http"
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
*:* )
|
||||||
|
HOST=$(printf "%s\n" "$1"| cut -d : -f 1)
|
||||||
|
PORT=$(printf "%s\n" "$1"| cut -d : -f 2)
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
-v | --version)
|
||||||
|
echo $VERSION
|
||||||
|
exit
|
||||||
|
;;
|
||||||
|
-q | --quiet)
|
||||||
|
QUIET=1
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
-q-*)
|
||||||
|
QUIET=0
|
||||||
|
echoerr "Unknown option: $1"
|
||||||
|
usage 1
|
||||||
|
;;
|
||||||
|
-q*)
|
||||||
|
QUIET=1
|
||||||
|
result=$1
|
||||||
|
shift 1
|
||||||
|
set -- -"${result#-q}" "$@"
|
||||||
|
;;
|
||||||
|
-t | --timeout)
|
||||||
|
TIMEOUT="$2"
|
||||||
|
shift 2
|
||||||
|
;;
|
||||||
|
-t*)
|
||||||
|
TIMEOUT="${1#-t}"
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
--timeout=*)
|
||||||
|
TIMEOUT="${1#*=}"
|
||||||
|
shift 1
|
||||||
|
;;
|
||||||
|
--)
|
||||||
|
shift
|
||||||
|
break
|
||||||
|
;;
|
||||||
|
--help)
|
||||||
|
usage 0
|
||||||
|
;;
|
||||||
|
-*)
|
||||||
|
QUIET=0
|
||||||
|
echoerr "Unknown option: $1"
|
||||||
|
usage 1
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
QUIET=0
|
||||||
|
echoerr "Unknown argument: $1"
|
||||||
|
usage 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
if ! [ "$TIMEOUT" -ge 0 ] 2>/dev/null; then
|
||||||
|
echoerr "Error: invalid timeout '$TIMEOUT'"
|
||||||
|
usage 3
|
||||||
|
fi
|
||||||
|
|
||||||
|
case "$PROTOCOL" in
|
||||||
|
tcp)
|
||||||
|
if [ "$HOST" = "" ] || [ "$PORT" = "" ]; then
|
||||||
|
echoerr "Error: you need to provide a host and port to test."
|
||||||
|
usage 2
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
http)
|
||||||
|
if [ "$HOST" = "" ]; then
|
||||||
|
echoerr "Error: you need to provide a host to test."
|
||||||
|
usage 2
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
wait_for "$@"
|
||||||
Loading…
Reference in New Issue