]> git.itanic.dy.fi Git - emerge-timer/blobdiff - emerge-timer.py
Fix correct newline behaviour in python3
[emerge-timer] / emerge-timer.py
index 739f8f4c4faa3c31c0707c5370efc42e5584ab4f..d782b34996fac6c9b6c791a58a069adace19de77 100755 (executable)
@@ -9,12 +9,18 @@ LOGFILE = "/var/log/emerge.log"
 
 
 green_start = "\033[32m"
+red_start = "\033[1;31m"
 color_stop = "\033[m"
 
 QUIET = False
 
 PACKAGES = []
 
+def GREEN(string):
+    return green_start + string + color_stop
+
+def RED(string):
+    return red_start + string + color_stop
 
 
 class package:
@@ -22,7 +28,9 @@ class package:
         self.name = name
         self.version = version
         self.versions = []
-        self.emerge_time = 12*3600
+        self.emerge_time = "infinity"
+
+        self.plotData = []
 
 
     def add_version(self, version, emerge_time, emerge_date):
@@ -30,6 +38,19 @@ class package:
         self.versions.append((version, emerge_time, emerge_date))
 
 
+    def seek_versions(self, whatToSeek, Input):
+
+        if whatToSeek == "date":
+            for i in self.versions:
+                if i[1] == Input:
+                    return i[2]
+
+        if whatToSeek == "version":
+            for i in self.versions:
+                if i[1] == Input:
+                    return i[0]
+
+
     def average_time(self):
         """Return average time from class version list"""
         total_time = 0
@@ -50,48 +71,62 @@ class package:
         return total_time
 
 
-    def max_time(self):
-        """Return maximum time from class version list"""
-        self.versions.sort()
+    def min_max_time(self, setting):
+        """Return maximum or minimum time from class version list"""
+
+        emerge_times = []
+        for i in self.versions:
+            emerge_times.append(i[1])
 
-        return self.versions[len(self.versions)-1][1]
+        emerge_times.sort()
 
+        if setting == "max":
+            return emerge_times[-1]
+        elif setting == "min":
+            return emerge_times[0]
 
-    def min_time(self):
-        """Return minimum time from class version list"""
-        self.versions.sort()
 
-        return self.versions[0][1]
 
 
     def print_current_emerge(self):
         """Function used to print all the current emerge stuff"""
 
-        print("\t" + green_start + self.name + '-' + self.version +
-              color_stop + "\n\t current time: " + give_time(self.emerge_time) +
-              "\n\t average time: "),
+        print("\t" + GREEN(self.name + '-' + self.version) +
+              "\n\t current time: " + give_time(self.emerge_time))
+
+
+        if len(self.versions) == 1:
+            print("\t last time:   "),
+            print(give_time(self.average_time()), end='')
+
+        elif len(self.versions) > 1:
+            print("\t average time: ", end='')
+            print(give_time(self.average_time()), end='')
 
-        if len(self.versions) > 1:
-            print(give_time(self.average_time())),
         else:
-            print("unknown"),
+            print("\t average time: " + GREEN("unknown\n")),
+            return
+
+        print("\n\t " + '-'*45, end='')
 
-        print("\n\t " + '-'*45),
+        print("\n\t time to finish: ", end='')
 
-        finish_time = self.average_time() - self.emerge_time
+        if type(self.emerge_time) != str:
 
-        print("\n\t time to finish:"),
+            finish_time = self.average_time() - self.emerge_time
 
-        if finish_time > 0:
-            print(give_time(finish_time))
+            if finish_time > 0:
+                print(give_time(finish_time), "\n")
+            else:
+                print(GREEN("any time now"))
         else:
-            print(green_start + "any time now" + color_stop)
+            print(GREEN("unknown"))
         print
 
 
     def print_versions(self):
         """This prints the emerge times for different versions in the
-        normal operating mode of the script"""
+        'package' operating mode of the script"""
 
         if QUIET == False:
 
@@ -105,27 +140,38 @@ class package:
                 if len(give_time(p[1], True)) > time_length:
                     time_length = len(give_time(p[1], True))
 
+                # Create the data for plotting in the format (emerge time, emerge date)
+                self.plotData.append((p[1], p[2]))
+
+            dots =  (version_length + time_length + len(self.name)
+                     + len(give_date(self.versions[0][2])) + 14)
+
             for p in self.versions:
 
                 pad = time_length - len(give_time(p[1], True))
 
-                print('-'*90 + "\n" +
-                      green_start + self.name + (p[0]).ljust(version_length) +
-                      color_stop + "  >>>  " + (give_time(p[1])) + " "*pad +
-                      "  >>>  " + give_date(p[2]))
+                name = self.name
+                p_time = give_time(p[1])
+                p_date = give_date(p[2])
 
-        print('-'*90 + "\n" + "Package " + green_start +
-              self.name + color_stop + " emerged " +
-              str(len(self.versions)) + " times.")
+                print('-' * dots + "\n" +
+                      GREEN(name + (p[0]).ljust(version_length))
+                      + "  >>>  " + p_time + " "*pad + "  >>>  " + p_date)
+
+        print("\n" + "Package " + GREEN(self.name) + " emerged"),
+
+        if len(self.versions) > 1:
+            print(str(len(self.versions)) + " times.\n")
+        elif len(self.versions) == 1:
+            print("once.\n")
 
-        print
 
 
     def print_pretended_times(self):
         """This is used the print all the pretended times"""
 
         if QUIET == False:
-            print("\t" + green_start + self.name + '-' + self.version + color_stop),
+            print("\t" + GREEN(self.name + '-' + self.version)),
 
         if len(self.versions) > 1:
             aver_time = self.average_time()
@@ -143,16 +189,55 @@ class package:
 
 
     def print_min_max_ave(self):
-        maxi = self.max_time()
-        mini = self.min_time()
+
+        if len(self.versions) == 1:
+            return
+
+        maxi = self.min_max_time("max")
+        mini = self.min_max_time("min")
         average = self.average_time()
         total = self.total_time()
 
-        print("Max time: \t" + give_time(maxi) +
-              "\nMin time: \t" + give_time(mini) +
-              "\nAverage time: \t" + give_time(average) +
-              "\nIn total spent " + give_time(total) +
-              " emerging " + green_start + self.name + color_stop)
+        print("-"*20 +
+              "\nMaximum emerge time:\n" +
+              give_time(maxi) + "for version " +
+              GREEN(self.seek_versions("version", maxi).lstrip("-"))
+              + " on " +
+              give_date(self.seek_versions("date", maxi)))
+
+        print("-"*20 +
+              "\nMinimum emerge time:\n" +
+              give_time(mini) + "for version " +
+              GREEN(self.seek_versions("version", mini).lstrip("-"))
+              + " on " +
+              give_date(self.seek_versions("date", mini)))
+
+        print("-"*20 +
+              "\nAverage emerge time:\n" +
+              give_time(average))
+
+        print("-"*20 +
+              "\nIn total spent:\n" +
+              give_time(total))
+
+
+    def plotToScreen(self):
+        dates = []
+        times = []
+
+        for i in self.plotData:
+            dates.append(datetime.date.fromtimestamp(i[1]))
+            times.append(i[0])
+
+        fig = plt.figure()
+
+        plt.plot_date(dates, times, xdate=True, ydate=False)
+
+        plt.ylabel("Emerge time [s]")
+        plt.suptitle(self.name)
+        plt.grid()
+        fig.autofmt_xdate()
+        plt.show()
 
 
 
@@ -167,10 +252,14 @@ def give_time(time, nocolor=False):
         green_start = ""
         color_stop = ""
 
-    days = time/(3600*24)
-    hours = (days - int(days))*24
-    minutes = (hours - int(hours))*60
-    seconds = (minutes - int(minutes))*60
+    if type(time) == str:
+        return(GREEN("unknown"))
+
+
+    days = time/(3600.0*24.0)
+    hours = (days - int(days))*24.0
+    minutes = (hours - int(hours))*60.0
+    seconds = (minutes - int(minutes))*60.0
 
     days = int(days)
     hours = int(hours)
@@ -183,31 +272,24 @@ def give_time(time, nocolor=False):
     pseconds = str()
 
     if days > 0:
-        pdays = (green_start + str(days) +
-                 color_stop + " day ")
+        pdays = (GREEN(str(days)) + " day ")
         if days != 1:
-            pdays = (green_start + str(days) +
-                         color_stop + " days ")
+            pdays = (GREEN(str(days)) + " days ")
 
     if hours > 0:
-        phours = (green_start + str(hours) +
-                  color_stop + " hour ")
+        phours = (GREEN(str(hours)) + " hour ")
         if hours != 1:
-            phours = (green_start + str(hours) +
-                          color_stop + " hours ")
+            phours = (GREEN(str(hours)) + " hours ")
 
     if minutes > 0:
-        pminutes = (green_start + str(minutes) +
-                    color_stop + " minute ")
+        pminutes = (GREEN(str(minutes)) + " minute ")
         if minutes != 1:
-            pminutes = (green_start + str(minutes) +
-                        color_stop + " minutes ")
+            pminutes = (GREEN(str(minutes)) + " minutes ")
 
-    pseconds = (green_start + str(seconds) +
-                color_stop + " second ")
-    if seconds != 1:
-        pseconds = (green_start + str(seconds) +
-                    color_stop + " seconds ")
+    if seconds > 0:
+        pseconds = (GREEN(str(seconds)) + " second ")
+        if seconds != 1:
+            pseconds = (GREEN(str(seconds)) + " seconds ")
 
     if nocolor == True:
         green_start = "\033[32m"
@@ -233,7 +315,7 @@ def open_log():
     try:
         f = open(LOGFILE, 'r')
     except IOError as detail:
-        print detail
+        print(detail)
         sys.exit(1)
 
     return f
@@ -270,6 +352,37 @@ def search_log_for_package(package_class):
 
 
 
+def search_log_for_all_packages():
+    """Goes through the emerge.log and lists all packages in there"""
+
+    log = open_log()
+
+    all_packages = []
+
+    total_emerge_time = 0
+    emerge_number = 0
+
+    for line in log:
+        if ((">>>" in line) and ("emerge" in line)):
+            pack = line.partition(')')[2].strip().partition(' ')[0]
+            start_time = float(line.partition(':')[0])
+
+            all_packages.append((pack, start_time))
+
+        elif ((":::" in line) and ("completed emerge" in line)):
+            for p in all_packages:
+                if p[0] in line:
+                    print("\t" + give_date(p[1]) + " >>> " + GREEN(p[0]))
+
+                    emerge_number += 1
+
+                    all_packages.pop(all_packages.index(p))
+
+
+    print("\n" + GREEN(str(emerge_number)) + " emerges in total found.")
+
+
+
 def get_package(name):
     """Take the user-input package name and search for it
     in PORTDIR. """
@@ -365,13 +478,16 @@ def list_emerge_processes():
 
 
     if len(PACKAGES) == 0:
-        print "No current emerge process found."
-        sys.exit(0)
+        print("No current emerge process found.")
+
+        return 1
 
 
     for line in f:
         if ((">>>" in line) and ("emerge" in line)):
             for p in PACKAGES:
+                difference = 0
+
                 if (p.name + '-' + p.version in line):
 
                     time = float(line.partition(' ')[0].strip(":"))
@@ -379,34 +495,79 @@ def list_emerge_processes():
                     timestamp = datetime.datetime.fromtimestamp(time)
                     difference = (now - timestamp).total_seconds()
 
-                    if difference < p.emerge_time:
+                    if ((p.emerge_time == "infinity") or
+                        (difference < p.emerge_time)):
+
                         p.emerge_time = difference
 
+    return 0
+
+
+def search_syncs():
+    f = open_log()
+
+    print("These emerge syncs found")
+    print("\tDate                    Server")
+    print("\t------------------------------")
+
+    a = 0
+    for line in f:
+        if "=== Sync completed with" in line:
+            time = float(line.partition(' ')[0].strip(":"))
+            server = line.rpartition(' ')[2]
+
+            print("\t" + GREEN(give_date(time)) +
+                  " === " + server),
+            a += 1
+
+    print("\n" + GREEN(str(a)) + " emerge syncs found. ")
 
 
 def main(status, user_package=None):
-    """Main function. Hanlde all the different modes of operation."""
+    try:
+        _main(status, user_package)
+    except IOError:
+        sys.exit()
+
+
+def _main(status, user_package=None):
+    """Main function. Handle all the different modes of operation."""
 
     if status == "package":
-        user_package = get_package(user_package)
+        for p in user_package:
+            pack = get_package(p)
 
-        pack = package(user_package)
+            pack = package(pack)
 
-        search_log_for_package(pack)
+            search_log_for_package(pack)
 
-        if len(pack.versions) != 0:
-            pack.print_versions()
-            pack.print_min_max_ave()
+            if len(pack.versions) != 0:
+                pack.print_versions()
+                pack.print_min_max_ave()
 
-        else:
-            print("Package " + green_start + pack.name +
-                  color_stop + " has never been emerged.")
+                if matplotWorking:
+                    pack.plotToScreen()
+
+            else:
+                print("Package " + GREEN(pack.name)
+                      + " has never been emerged.")
+
+
+    elif status == "sync":
+        search_syncs()
+        return
+
+
+    elif status == "list":
+        search_log_for_all_packages()
+        return
 
 
     elif status == "current":
-        list_emerge_processes()
+        if list_emerge_processes():
+            return
 
-        print "Currently emerging:"
+        print("Currently emerging:")
 
         for p in PACKAGES:
             search_log_for_package(p)
@@ -416,7 +577,7 @@ def main(status, user_package=None):
     elif status == "pretended":
         list_pretended()
 
-        print "This is how long these packages would take to emerge"
+        print("This is how long these packages would take to emerge")
 
         total_pretended_time = 0
 
@@ -427,8 +588,8 @@ def main(status, user_package=None):
 
             print
 
-        print("Total emerge time of " + green_start + str(len(PACKAGES)) +
-              color_stop + " package(s): "+ give_time(total_pretended_time))
+        print("Total emerge time of " + GREEN(str(len(PACKAGES)))
+              + " package(s): "+ give_time(total_pretended_time))
 
 
 def usage():
@@ -439,11 +600,16 @@ Calculate emerge times from emerge log.
 Options:
 \t-c, --current \t Show time until currently compiling package finishes
 \t-p, --pretended  Calculate compile time from piped 'emerge -p' output
+\t-l, --list \t List all emerged packages
+\t-s, --sync \t Show emerge sync history
 \t-h, --help \t Show this helpscreen
 \t-q, --quiet \t Be less verbose
-\t--no-color \t Use colorless output"""
+\t--no-color \t Use colorless output
+\t--plot \t\t Plot emerge times into a 2D scatter plot
+\t\t\t (needs matlibplot installed for this to work)"""
 
-    print usage
+
+    print(usage)
 
     sys.exit(0)
 
@@ -452,8 +618,9 @@ if __name__ == "__main__":
 
     # Set the default mode as "package"
     mode = "package"
-    input_package = None
-
+    input_packages = None
+    simulation = False
+    matplotWorking = False
 
     for arg in sys.argv[1:]:
 
@@ -466,17 +633,41 @@ if __name__ == "__main__":
         if arg == "-h" or arg == "--help":
             usage()
 
+        if arg == "-l" or arg == "--list":
+            mode = "list"
+
+        if arg == "-s" or arg == "--sync":
+            mode = "sync"
+
         if arg == "-q" or arg == "--quiet":
             QUIET = True
 
+            sys.argv.pop(sys.argv.index(arg))
+
+        if arg == "--plot":
+            try:
+                import matplotlib.pyplot as plt
+                matplotWorking = True
+            except ImportError:
+                print(RED("Cannot initialize matplotlib!"))
+                print(RED("Check that you have properly installed matplotlib.\n"))
+                matplotWorking = False
+
+            sys.argv.pop(sys.argv.index(arg))
+
         if arg == "--no-color":
             green_start = ""
             color_stop = ""
 
+            sys.argv.pop(sys.argv.index(arg))
+
+        if arg == "--simulate":
+            simulation = True
+
 
     if len(sys.argv) > 1:
-        input_package = sys.argv[1]
+        input_packages = sys.argv[1:]
     else:
         usage()
 
-    main(mode, input_package)
+    main(mode, input_packages)