Gecko [ aodai.devttt.com ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
.my.cnf.4370	0 B	-rw-r--r--
.mysql.4370	0 B	-rw-r--r--
Backupsv2.py	59.03 KB	-rw-r--r--
ClusterManager.py	17.68 KB	-rw-r--r--
CyberCPLogFileWriter.py	4.37 KB	-rw-r--r--
CyberPanelUpgrade.py	3.64 KB	-rw-r--r--
DockerSites.py	61.86 KB	-rw-r--r--
IncScheduler.py	88.07 KB	-rw-r--r--
SwitchOldAliasToNew.py	2.4 KB	-rw-r--r--
__init__.py	0 B	-rw-r--r--
acl.py	50.61 KB	-rw-r--r--
acl.py.backup_dns_fix	48.95 KB	-rw-r--r--
adminPass.py	2.41 KB	-rw-------
alias.py	647 B	-rw-r--r--
apiAccess.py	418 B	-rw-r--r--
applicationInstaller.py	352.05 KB	-rw-r--r--
backupSchedule.py	22.21 KB	-rw-r--r--
backupScheduleLocal.py	5.31 KB	-rw-r--r--
backupUtilities.py	110.15 KB	-rw-r--r--
bandwidthReset.py	7.25 KB	-rw-r--r--
cPanelImporter.py	41.33 KB	-rw-r--r--
childDomain.py	1.25 KB	-rw-r--r--
cronUtil.py	5.34 KB	-rw-r--r--
csf.py	48.63 KB	-rw-r--r--
customACME.py	51.55 KB	-rw-r--r--
dnsUtilities.py	38.7 KB	-rw-r--r--
domain.xml	0 B	-rw-r--r--
emergency_2fa_disable.py	1.24 KB	-rw-r--r--
filesPermsUtilities.py	6.17 KB	-rw-r--r--
filesPermsUtilities.py.bak	6.17 KB	-rw-r--r--
findBWUsage.py	4.6 KB	-rw-r--r--
firewallUtilities.py	8.18 KB	-rw-r--r--
ftpUtilities.py	7.56 KB	-rw-r--r--
getSystemInformation.py	5.29 KB	-rw-r--r--
hashPassword.py	1.22 KB	-rw-r--r--
httpProc.py	4.81 KB	-rw-r--r--
installUtilities.py	19.04 KB	-rw-r--r--
letsEncrypt.py	0 B	-rw-r--r--
mailUtilities.py	104.79 KB	-rw-r--r--
modSec.py	26.71 KB	-rw-r--r--
mysqlUtilities.py	41.46 KB	-rw-r--r--
phpUtilities.py	19.8 KB	-rw-r--r--
phpmyadminsignin.php	2.08 KB	-rw-r--r--
pluginManagerGlobal.py	813 B	-rw-r--r--
processUtilities.py	22.42 KB	-rw-r--r--
randomPassword.py	200 B	-rw-r--r--
rebuildQuotas.py	3.29 KB	-rw-r--r--
remoteBackup.py	14.81 KB	-rw-r--r--
remoteTransferUtilities.py	16.43 KB	-rw-r--r--
renew.py	5.64 KB	-rwxr-xr-x
restoreMeta.py	7.65 KB	-rw-r--r--
serverLogs.py	735 B	-rw-r--r--
sslUtilities.py	49.26 KB	-rw-r--r--
sslv2.py	26.18 KB	-rw-r--r--
test.py	0 B	-rw-r--r--
test1.py	0 B	-rw-r--r--
tuning.py	18.08 KB	-rw-r--r--
upgrade.py	213.62 KB	-rw-r--r--
upgrade.py.bak	97.03 KB	-rw-r--r--
upgradeCritical.py	1.15 KB	-rwx------
vhost.py	47 KB	-rw-r--r--
vhostConfs.py	14.41 KB	-rw-r--r--
virtualHostUtilities.py	92.31 KB	-rw-r--r--

Code Editor : bandwidthReset.py

#!/usr/local/CyberCP/bin/python
import sys
sys.path.append('/usr/local/CyberCP')
import os
import json
from plogical import CyberCPLogFileWriter as logging
from websiteFunctions.models import Websites, ChildDomains

class BandwidthReset:
    """
    Bandwidth reset utility for CyberPanel
    Resets monthly bandwidth usage for all websites and child domains
    """
    
    @staticmethod
    def resetWebsiteBandwidth():
        """
        Reset bandwidth usage for all websites and child domains
        """
        try:
            logging.CyberCPLogFileWriter.writeToFile("Starting monthly bandwidth reset...")
            
            # Reset main websites
            websites = Websites.objects.all()
            reset_count = 0
            
            for website in websites:
                try:
                    # Load current config
                    try:
                        config = json.loads(website.config)
                    except:
                        config = {}
                    
                    # Reset bandwidth data
                    config['bwInMB'] = 0
                    config['bwUsage'] = 0
                    
                    # Save updated config
                    website.config = json.dumps(config)
                    website.save()
                    
                    reset_count += 1
                    logging.CyberCPLogFileWriter.writeToFile(f"Reset bandwidth for website: {website.domain}")
                    
                except Exception as e:
                    logging.CyberCPLogFileWriter.writeToFile(f"Error resetting bandwidth for website {website.domain}: {str(e)}")
            
            # Reset child domains
            child_domains = ChildDomains.objects.all()
            
            for child in child_domains:
                try:
                    # Load current config
                    try:
                        config = json.loads(child.config)
                    except:
                        config = {}
                    
                    # Reset bandwidth data
                    config['bwInMB'] = 0
                    config['bwUsage'] = 0
                    
                    # Save updated config
                    child.config = json.dumps(config)
                    child.save()
                    
                    reset_count += 1
                    logging.CyberCPLogFileWriter.writeToFile(f"Reset bandwidth for child domain: {child.domain}")
                    
                except Exception as e:
                    logging.CyberCPLogFileWriter.writeToFile(f"Error resetting bandwidth for child domain {child.domain}: {str(e)}")
            
            # Clean up bandwidth metadata files
            BandwidthReset.cleanupBandwidthMetadata()
            
            logging.CyberCPLogFileWriter.writeToFile(f"Monthly bandwidth reset completed. Reset {reset_count} domains.")
            return True
            
        except Exception as e:
            logging.CyberCPLogFileWriter.writeToFile(f"Error in monthly bandwidth reset: {str(e)}")
            return False
    
    @staticmethod
    def cleanupBandwidthMetadata():
        """
        Clean up bandwidth metadata files
        """
        try:
            import glob
            
            # Clean up main bandwidth metadata files
            metadata_files = glob.glob("/home/cyberpanel/*.bwmeta")
            for file_path in metadata_files:
                try:
                    # Reset the metadata file to 0 usage
                    with open(file_path, 'w') as f:
                        f.write("0\n0\n")
                    os.chmod(file_path, 0o600)
                    logging.CyberCPLogFileWriter.writeToFile(f"Reset metadata file: {file_path}")
                except Exception as e:
                    logging.CyberCPLogFileWriter.writeToFile(f"Error resetting metadata file {file_path}: {str(e)}")
            
            # Clean up domain-specific bandwidth metadata files
            domain_metadata_files = glob.glob("/home/*/logs/bwmeta")
            for file_path in domain_metadata_files:
                try:
                    # Reset the metadata file to 0 usage
                    with open(file_path, 'w') as f:
                        f.write("0\n0\n")
                    os.chmod(file_path, 0o600)
                    logging.CyberCPLogFileWriter.writeToFile(f"Reset domain metadata file: {file_path}")
                except Exception as e:
                    logging.CyberCPLogFileWriter.writeToFile(f"Error resetting domain metadata file {file_path}: {str(e)}")
                    
        except Exception as e:
            logging.CyberCPLogFileWriter.writeToFile(f"Error cleaning up bandwidth metadata: {str(e)}")
    
    @staticmethod
    def resetSpecificDomain(domain_name):
        """
        Reset bandwidth for a specific domain
        """
        try:
            # Try to find as main website
            try:
                website = Websites.objects.get(domain=domain_name)
                try:
                    config = json.loads(website.config)
                except:
                    config = {}
                
                config['bwInMB'] = 0
                config['bwUsage'] = 0
                website.config = json.dumps(config)
                website.save()
                
                logging.CyberCPLogFileWriter.writeToFile(f"Reset bandwidth for website: {domain_name}")
                return True
                
            except Websites.DoesNotExist:
                pass
            
            # Try to find as child domain
            try:
                child = ChildDomains.objects.get(domain=domain_name)
                try:
                    config = json.loads(child.config)
                except:
                    config = {}
                
                config['bwInMB'] = 0
                config['bwUsage'] = 0
                child.config = json.dumps(config)
                child.save()
                
                logging.CyberCPLogFileWriter.writeToFile(f"Reset bandwidth for child domain: {domain_name}")
                return True
                
            except ChildDomains.DoesNotExist:
                logging.CyberCPLogFileWriter.writeToFile(f"Domain not found: {domain_name}")
                return False
                
        except Exception as e:
            logging.CyberCPLogFileWriter.writeToFile(f"Error resetting bandwidth for domain {domain_name}: {str(e)}")
            return False

def main():
    import argparse
    
    parser = argparse.ArgumentParser(description='CyberPanel Bandwidth Reset Utility')
    parser.add_argument('--reset-all', action='store_true', help='Reset bandwidth for all domains')
    parser.add_argument('--domain', help='Reset bandwidth for specific domain')
    parser.add_argument('--cleanup-metadata', action='store_true', help='Clean up bandwidth metadata files only')
    
    args = parser.parse_args()
    
    if args.reset_all:
        BandwidthReset.resetWebsiteBandwidth()
    elif args.domain:
        BandwidthReset.resetSpecificDomain(args.domain)
    elif args.cleanup_metadata:
        BandwidthReset.cleanupBandwidthMetadata()
    else:
        print("Please specify an action: --reset-all, --domain <domain_name>, or --cleanup-metadata")
        sys.exit(1)

if __name__ == "__main__":
    main()

${this.title}

Code Editor : bandwidthReset.py