Untitled

 avatar
unknown
plain_text
2 years ago
2.9 kB
2
Indexable
def purge_logs():
    vm_details = get_vm_details()

    for vm in vm_details:
        vm_id = vm['vm_id']
        vm_ip = vm['vm_ip']
        syslog_file = vm['syslog_file_location']
        retention_period = get_retention_period(vm_id)
        bookmark_index = get_bookmark_index(vm_ip)

        print("VM:", vm_id)
        print("Syslog File Location:", syslog_file)
        print("Retention Period:", retention_period)
        print("Bookmark Index:", bookmark_index)
        print("Purging Logs...")

        try:
            bookmark_line = linecache.getline(syslog_file, int(bookmark_index)).strip()
            print("Bookmark Line:", bookmark_line)
        except Exception as e:
            print("Exception while reading line from file:", e)
            return None

        if not bookmark_line:
            print("Bookmark line not found.")
            continue

        bookmark_timestamp = get_timestamp_from_line(bookmark_line)
        print("Bookmark TImestamp : ", bookmark_timestamp)

        # Calculate the purging date (retention_period days before the bookmark timestamp)
        bookmark_date = datetime.datetime.strptime(bookmark_timestamp, "%Y-%m-%dT%H:%M:%S+00:00").date()
        #bookmark_date=datetime.date(bookmark_date)
        print("Bookmark Date : ", bookmark_date)
        
        current_date= datetime.datetime.now().date()
        print("Current Date:",current_date)

        purging_date = current_date - datetime.timedelta(days=retention_period)
        print("Purging Date : ", purging_date)

        # Perform the log purging
        lines_purged = 0  # Counter for purged lines
        temp_file = syslog_file + ".tmp"  # Create a temporary file
        
        purge_mode = False  # Flag to indicate if purging is needed
        
        if bookmark_date > purging_date :
            purge_mode=True
            with open(syslog_file, "r") as input_file, open(temp_file, "w") as output_file:
                for line in input_file:
                    timestamp = get_timestamp_from_line(line)
                    print(timestamp)
                    line_date = datetime.datetime.strptime(bookmark_timestamp, "%Y-%m-%dT%H:%M:%S+00:00").date()
                    print(line_date)
                    if line_date > purging_date:
                        output_file.write(line)
                    else:
                        lines_purged += 1
        else:
            print("Purging is not needed.")
                    

        # Replace the original file with the temporary file
        shutil.move(temp_file, syslog_file)

        print("Purging Completed")
        print(f"Lines Purged: {lines_purged}")
        print("Printing First Two Lines:")
        with open(syslog_file, "r") as file:
            print(file.readline())
            print(file.readline())
        print()

purge_logs()