-
Notifications
You must be signed in to change notification settings - Fork 1
/
allowed.py
61 lines (50 loc) · 2.16 KB
/
allowed.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
import re, os, sys
p_name = sys.argv[1]
SETTINGS_PATH = f'/opt/projects/{p_name}/{p_name}/{p_name}/settings'
BASE_PATH = os.getcwd()
BACK_PATH = BASE_PATH + f'/backup_temp/opt/domens'
TEMP1 = 'ALLOWED_HOSTS = [\'*\']'
TEMP2 = 'ALLOWED_HOSTS = ['
TEMP3 = 'ALLOWED_HOSTS = [\'localhost\']'
start_dev = []
start_prod = []
dev_result = []
prod_result = []
with open(BACK_PATH + f'/{p_name}_domens.txt', 'r') as domens_txt:
domens = domens_txt.readlines()
# СОСТАВЛЕНИЕ ЧАСТЕЙ dev.py + домены
with open(SETTINGS_PATH + '/dev.py', 'r') as dev:
allow_host_dev = dev.readline()
while 'ALLOWED_HOSTS' not in allow_host_dev:
start_dev.append(allow_host_dev)
allow_host_dev = dev.readline()
finish__dev = dev.readlines()
for domen in domens:
domen = domen.strip()
dev_result.append(' \'' + domen + '\',')
# СОСТАВЛЕНИЕ ЧАСТЕЙ prod.py + домены
with open(SETTINGS_PATH + '/prod.py', 'r') as prod:
allow_host_prod = prod.readline()
while 'ALLOWED_HOSTS' not in allow_host_prod:
start_prod.append(allow_host_prod)
allow_host_prod = prod.readline()
finish_prod = prod.readlines()
for domen in domens:
domen = domen.strip()
prod_result.append(' \'' + domen + '\',')
# Запись измененного DEV.PY
with open(SETTINGS_PATH + '/dev.py', 'w') as dev:
if allow_host_dev.strip() == TEMP1:
dev.write(''.join(start_dev) + 'ALLOWED_HOSTS = [\n' + '\n'.join(dev_result)\
+ '\n]\n' + ''.join(finish__dev))
elif allow_host_dev.strip() == TEMP2:
dev.write(''.join(start_dev) + 'ALLOWED_HOSTS = [\n' + '\n'.join(dev_result)\
+ '\n' + ''.join(finish__dev))
# Запись измененного PROD.PY
with open(SETTINGS_PATH + '/prod.py', 'w') as prod:
if allow_host_prod.strip() == TEMP3:
prod.write(''.join(start_prod) + 'ALLOWED_HOSTS = [\n' + '\n'.join(prod_result)\
+ '\n]\n' + ''.join(finish_prod))
elif allow_host_prod.strip() == TEMP2:
prod.write(''.join(start_prod) + 'ALLOWED_HOSTS = [\n' + '\n'.join(prod_result)\
+ '\n' + ''.join(finish_prod))