Use coverage package for rounding the percentage (dbrgn#6)

Author: simonfagerholm

coverage_badge/__main__.py

@@ -46,14 +46,31 @@ def write(self, *args, **kwargs):
         pass
 
 
+class Precision(coverage.results.Numbers):
+    """
+    A class for using the percentage rounding of the main coverage package,
+    with any percentage.
+
+    _Precision.pc_covered_str is the string format of the percentage.
+    """
+    def __init__(self, percent):
+        self.percent = percent
+
+    @property
+    def pc_covered(self):
+        """overriding the percentage calculation of the parent class"""
+        return self.percent
+
+
 def get_total():
     """
     Return the rounded total as properly rounded string.
     """
     cov = coverage.Coverage()
     cov.load()
     total = cov.report(file=Devnull())
-    return '{0:.0f}'.format(total)
+
+    return Precision(total).pc_covered_str
 
 
 def get_color(total):