2022-04-12 09:45:10 +00:00
# Generated by Django 3.1.14 on 2022-04-12 07:39
2022-05-07 08:20:12 +00:00
import common . db . fields
2022-04-12 09:45:10 +00:00
import django . core . validators
from django . db import migrations , models
import django . db . models . deletion
import uuid
from django . conf import settings
def migrate_endpoints ( apps , schema_editor ) :
2022-04-18 07:17:28 +00:00
Endpoint = apps . get_model ( " terminal " , " Endpoint " )
# migrate default
default_data = {
2022-04-12 09:45:10 +00:00
' id ' : ' 00000000-0000-0000-0000-000000000001 ' ,
' name ' : ' Default ' ,
' host ' : ' ' ,
' https_port ' : 0 ,
' http_port ' : 0 ,
' created_by ' : ' System '
}
2022-04-19 02:34:17 +00:00
Endpoint . objects . create ( * * default_data )
2022-04-12 09:45:10 +00:00
2022-06-06 10:04:11 +00:00
if not settings . TERMINAL_RAZOR_ENABLED :
2022-04-18 07:17:28 +00:00
return
# migrate xrdp
xrdp_addr = settings . TERMINAL_RDP_ADDR
if ' : ' in xrdp_addr :
host , rdp_port = xrdp_addr . strip ( ) . split ( ' : ' )
else :
host , rdp_port = xrdp_addr , 3389
host = host . strip ( )
if host in [ ' localhost ' , ' 127.0.0.1 ' ] :
host = ' '
if not host :
return
if isinstance ( rdp_port , str ) and rdp_port . isdigit ( ) :
rdp_port = int ( rdp_port )
elif isinstance ( rdp_port , int ) and ( 0 < = rdp_port < = 65535 ) :
rdp_port = rdp_port
else :
rdp_port = 3389
xrdp_data = {
2022-06-06 10:04:11 +00:00
' name ' : ' Razor ' ,
2022-04-18 07:17:28 +00:00
' host ' : host ,
' https_port ' : 0 ,
' http_port ' : 0 ,
' ssh_port ' : 0 ,
' rdp_port ' : rdp_port ,
' mysql_port ' : 0 ,
' mariadb_port ' : 0 ,
' postgresql_port ' : 0 ,
' created_by ' : ' System '
}
xrdp_endpoint = Endpoint . objects . create ( * * xrdp_data )
EndpointRule = apps . get_model ( " terminal " , " EndpointRule " )
xrdp_rule_data = {
2022-06-06 10:04:11 +00:00
' name ' : ' Razor ' ,
2022-04-18 07:17:28 +00:00
' ip_group ' : [ ' * ' ] ,
' priority ' : 20 ,
' endpoint ' : xrdp_endpoint ,
' created_by ' : ' System '
}
EndpointRule . objects . create ( * * xrdp_rule_data )
2022-04-12 09:45:10 +00:00
class Migration ( migrations . Migration ) :
dependencies = [
( ' terminal ' , ' 0047_auto_20220302_1951 ' ) ,
]
operations = [
migrations . CreateModel (
name = ' Endpoint ' ,
fields = [
( ' created_by ' , models . CharField ( blank = True , max_length = 32 , null = True , verbose_name = ' Created by ' ) ) ,
( ' updated_by ' , models . CharField ( blank = True , max_length = 32 , null = True , verbose_name = ' Updated by ' ) ) ,
( ' date_created ' , models . DateTimeField ( auto_now_add = True , null = True , verbose_name = ' Date created ' ) ) ,
( ' date_updated ' , models . DateTimeField ( auto_now = True , verbose_name = ' Date updated ' ) ) ,
( ' id ' , models . UUIDField ( default = uuid . uuid4 , primary_key = True , serialize = False ) ) ,
2022-04-19 02:34:17 +00:00
( ' name ' , models . CharField ( max_length = 128 , unique = True , verbose_name = ' Name ' ) ) ,
( ' host ' , models . CharField ( max_length = 256 , verbose_name = ' Host ' , blank = True ) ) ,
2022-05-07 08:20:12 +00:00
( ' https_port ' , common . db . fields . PortField ( default = 443 , validators = [ django . core . validators . MinValueValidator ( 0 ) , django . core . validators . MaxValueValidator ( 65535 ) ] , verbose_name = ' HTTPS Port ' ) ) ,
( ' http_port ' , common . db . fields . PortField ( default = 80 , validators = [ django . core . validators . MinValueValidator ( 0 ) , django . core . validators . MaxValueValidator ( 65535 ) ] , verbose_name = ' HTTP Port ' ) ) ,
( ' ssh_port ' , common . db . fields . PortField ( default = 2222 , validators = [ django . core . validators . MinValueValidator ( 0 ) , django . core . validators . MaxValueValidator ( 65535 ) ] , verbose_name = ' SSH Port ' ) ) ,
( ' rdp_port ' , common . db . fields . PortField ( default = 3389 , validators = [ django . core . validators . MinValueValidator ( 0 ) , django . core . validators . MaxValueValidator ( 65535 ) ] , verbose_name = ' RDP Port ' ) ) ,
( ' mysql_port ' , common . db . fields . PortField ( default = 33060 , validators = [ django . core . validators . MinValueValidator ( 0 ) , django . core . validators . MaxValueValidator ( 65535 ) ] , verbose_name = ' MySQL Port ' ) ) ,
( ' mariadb_port ' , common . db . fields . PortField ( default = 33061 , validators = [ django . core . validators . MinValueValidator ( 0 ) , django . core . validators . MaxValueValidator ( 65535 ) ] , verbose_name = ' MariaDB Port ' ) ) ,
( ' postgresql_port ' , common . db . fields . PortField ( default = 54320 , validators = [ django . core . validators . MinValueValidator ( 0 ) , django . core . validators . MaxValueValidator ( 65535 ) ] , verbose_name = ' PostgreSQL Port ' ) ) ,
2022-04-12 09:45:10 +00:00
( ' comment ' , models . TextField ( blank = True , default = ' ' , verbose_name = ' Comment ' ) ) ,
] ,
options = {
' verbose_name ' : ' Endpoint ' ,
' ordering ' : ( ' name ' , ) ,
} ,
) ,
migrations . CreateModel (
name = ' EndpointRule ' ,
fields = [
( ' created_by ' , models . CharField ( blank = True , max_length = 32 , null = True , verbose_name = ' Created by ' ) ) ,
( ' updated_by ' , models . CharField ( blank = True , max_length = 32 , null = True , verbose_name = ' Updated by ' ) ) ,
( ' date_created ' , models . DateTimeField ( auto_now_add = True , null = True , verbose_name = ' Date created ' ) ) ,
( ' date_updated ' , models . DateTimeField ( auto_now = True , verbose_name = ' Date updated ' ) ) ,
( ' id ' , models . UUIDField ( default = uuid . uuid4 , primary_key = True , serialize = False ) ) ,
( ' name ' , models . CharField ( max_length = 128 , unique = True , verbose_name = ' Name ' ) ) ,
( ' ip_group ' , models . JSONField ( default = list , verbose_name = ' IP group ' ) ) ,
( ' priority ' , models . IntegerField ( help_text = ' 1-100, the lower the value will be match first ' , unique = True , validators = [ django . core . validators . MinValueValidator ( 1 ) , django . core . validators . MaxValueValidator ( 100 ) ] , verbose_name = ' Priority ' ) ) ,
( ' comment ' , models . TextField ( blank = True , default = ' ' , verbose_name = ' Comment ' ) ) ,
( ' endpoint ' , models . ForeignKey ( blank = True , null = True , on_delete = django . db . models . deletion . SET_NULL , related_name = ' rules ' , to = ' terminal.endpoint ' , verbose_name = ' Endpoint ' ) ) ,
] ,
options = {
' verbose_name ' : ' Endpoint rule ' ,
' ordering ' : ( ' priority ' , ' name ' ) ,
} ,
) ,
migrations . RunPython ( migrate_endpoints ) ,
]