-
Notifications
You must be signed in to change notification settings - Fork 2
/
update.py
62 lines (54 loc) · 2.21 KB
/
update.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
62
import os
import subprocess
#subprocess.call(['chmod', '0777', '/etc/squid/squid.conf'])
def add_port(username,port_num):
file = open('/etc/squid/port.conf','a')
file.write("\n"+"acl Safe_ports port "+str(port_num))
file.close()
subprocess.call('sudo systemctl reload squid.service',shell=True)
def add_mac(username,mac):
file = open('/etc/squid/config_files/'+username+'/'+username+'_mac.lst','a')
file.write("\n"+str(mac))
file.close()
subprocess.call('sudo systemctl reload squid.service',shell=True)
def add_website(username,website):
file = open('/etc/squid/config_files/'+username+'/'+username+'_website.lst','a')
file.write("\n"+str(website))
file.close()
subprocess.call('sudo systemctl reload squid.service',shell=True)
def remove_mac(username,mac):
file = open('/etc/squid/config_files/'+username+'/'+username+'_mac.lst','r')
macsList = file.readlines()
file.close()
file = open('/etc/squid/config_files/'+username+'/'+username+'_mac.lst','w')
file.close()
for line in macsList:
file = open('/etc/squid/config_files/'+username+'/'+username+'_mac.lst','a')
if str(line.strip())!=mac and line.strip()!="":
file.write("\n"+str(line.strip()))
file.close()
subprocess.call('sudo systemctl reload squid.service',shell=True)
def remove_port(username,port):
file = open('/etc/squid/port.conf','r')
portsList = file.readlines()
file.close()
file = open('/etc/squid/port.conf','w')
file.close()
for line in portsList:
file = open('/etc/squid/port.conf','a')
if str(line.strip())!=("acl Safe_ports port "+str(port)) and line.strip()!="":
file.write("\n"+str(line.strip()))
file.close()
subprocess.call('sudo systemctl reload squid.service',shell=True)
def remove_website(username,website):
file = open('/etc/squid/config_files/'+username+'/'+username+'_website.lst','r')
websitesList = file.readlines()
file.close()
file = open('/etc/squid/config_files/'+username+'/'+username+'_website.lst','w')
file.close()
for line in websitesList:
file = open('/etc/squid/config_files/'+username+'/'+username+'_website.lst','a')
if str(line.strip())!=website and line.strip()!="":
file.write("\n"+str(line.strip()))
file.close()
subprocess.call('sudo systemctl reload squid.service',shell=True)