D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
proc
/
2
/
task
/
2
/
root
/
opt
/
dedrads
/
Filename :
exclude_rbl.py
back
Copy
#! /usr/lib/rads/venv/bin/python3 """ Written by Quintin Holmes """ import argparse import subprocess import os import re # add and remove domains to/from /etc/skiprbldomains def modify_domains(file_path, domains, action="add"): modified = False try: # read the existing domains with open(file_path, "r+", encoding="utf-8") as file: existing_domains = {line.strip() for line in file} # reset file pointer to the beginning for re-writing if needed file.seek(0) if action == "add": for domain in domains: if domain not in existing_domains: existing_domains.add(domain) modified = True # rewrite the file with updated domains if modified: file.writelines( f"{domain}\n" for domain in existing_domains ) # file.truncate() # Adjust file size to match new content elif action == "remove": initial_count = len(existing_domains) existing_domains.difference_update(domains) # Check if any domains were removed if len(existing_domains) < initial_count: modified = True print(f"Domains removed from {file_path}.") file.seek(0) file.writelines( f"{domain}\n" for domain in existing_domains ) file.truncate() else: print(f"No specified domains were found in {file_path}.") except PermissionError: print(f"Permission denied: Unable to modify {file_path}") return modified # restart Exim def restart_exim(): try: subprocess.run(["/scripts/buildeximconf"], check=True) subprocess.run(["systemctl", "restart", "exim"], check=True) print("Exim restarted.") except subprocess.CalledProcessError as e: print(f"Error restarting Exim: {e}") # looks for a cp user in userdomains, then adds all domains for the user to # skiprbldomains def find_user_domains(user, userdomains_path="/etc/userdomains"): domains = [] try: with open(userdomains_path, encoding="utf-8") as file: for line in file: domain, owner = line.strip().split(':') if owner.strip() == user: domains.append(domain.strip()) except PermissionError: print(f"Permission denied: Unable to read {userdomains_path}") return domains def child_accounts(reseller, file_path): accounts = [] with open('/etc/trueuserowners', encoding="utf-8") as file: for line in file: if re.search(f":{reseller}", line): accounts.append(line.split(':')[0]) reseller_domains = [] for account in accounts: with open('/etc/trueuserdomains', encoding="utf-8") as file: for line in file: if re.search(f":{account}$", line): reseller_domains.append(line.split(':')[0].strip()) if reseller_domains: modify_domains(file_path, reseller_domains, action="add") # Main function of the script that has argument setup def main(): parser = argparse.ArgumentParser( description="Modify /etc/skiprbldomains file." ) parser.add_argument( "-a", "--add", nargs='+', help="Add domains to the file.", metavar="DOMAIN", default=[], type=str, ) parser.add_argument( "-r", "--remove", nargs='+', help="Remove domains from the file.", metavar="DOMAIN", default=[], type=str, ) parser.add_argument( "-u", "--user", help="Add all domains owned by this user.", metavar="USERNAME", type=str, ) parser.add_argument( "-c", "--child", action="store_true", help="Add reseller child accounts.", ) args = parser.parse_args() file_path = "/etc/skiprbldomains" domains_added = False if not os.path.isfile(file_path): try: open(file_path, "a", encoding='utf-8').close() except PermissionError: print(f"Permission denied: Unable to create or modify {file_path}") return # add and remove actions if args.add: domains_added = modify_domains(file_path, args.add, action="add") if args.remove: domains_added = ( modify_domains(file_path, args.remove, action="remove") or domains_added ) # -- user is used, add their domains if args.user: user_domains = find_user_domains(args.user) if user_domains: domains_added = ( modify_domains(file_path, user_domains, action="add") or domains_added ) if args.child: child_accounts(args.user, file_path) domains_added = True if domains_added: restart_exim() else: print("No changes made.") if __name__ == "__main__": main()