Changeset 3002

Timestamp:

Sep 3, 2015, 1:16:58 PM (10 years ago)

Author:

Александър Шопов

Message:

Форматиран изход, да се оправят аргументи на командния ред

File:

• scripts/xml-stats/stats.py (modified) (4 diffs)

scripts/xml-stats/stats.py

@@ -9 +9 @@
 import sys
 
-titles = {'count': 'Номе',
-          'name': 'Име',
-          'branch': 'Клон',
-          'work':'Работ',
-          'untranslated': 'Непре',
-          'fuzzy': 'Мъгля',
-          'translated': 'Превед',
-          'total': 'Всичко'}
+Stat = col.namedtuple('Stat', ['count', 'name', 'branch', 'work',
+                               'untranslated','fuzzy', 'translated', 'total'])
+
+#titles = Stat(count='Номер', name='Име', branch='Клон', work='Работа',
+#              untranslated='Непреведени',  fuzzy='Мъгляви', translated='Преведени',
+#              total='Всичко')
+
+titles = Stat(count=('Number', 'rjust'), name=('Name', 'ljust'),
+              branch=('Branch', 'ljust'), work=('Work', 'ljust'),
+              untranslated=('Untranslated', 'ljust'),  fuzzy=('Fuzyy', 'rjust'),
+              translated=('Translated', 'rjust'), total=('All', 'rjust'))
+
 
 def get_versions():
@@ -28 +32 @@
     versions = [e.text for e in version_elements]
     return versions
+
 
 def get_stats(version):
@@ -54 +59 @@
             work = fuzzy + untranslated
             total = work + translated
-            s = {'name': name,
-                 'branch': stat_branch,
-                 'translated': translated,
-                 'fuzzy': fuzzy,
-                 'untranslated': untranslated,
-                 'work': work,
-                 'total': total} 
+            s = Stat(count=None, name=name, branch=stat_branch,
+                     translated=translated, fuzzy=fuzzy,
+                     untranslated=untranslated, work=work, total=total)
             results.append(s)
+    # sorts are stable
+    results.sort(key=lambda s: s.name)
+    results.sort(key=lambda s: s.fuzzy, reverse=True)
+    results.sort(key=lambda s: s.untranslated, reverse=True)
+    results.sort(key=lambda s: s.work, reverse=True)
+    for i, s in enumerate(results):
+        results[i] = s._replace(count=i+1)
     return results
 
+
 def stats_max_lengths_sums(stats):
-    string_keys = [ 'name', 'branch']
-    number_keys = ['translated', 'fuzzy', 'untranslated', 'work', 'total']
-    string_lengths = {k: 0 for k in string_keys}
-    number_sums = {k: 0 for k in number_keys}
+    name = 0
+    branch = 0
+    sum_translated = 0
+    sum_fuzzy = 0
+    sum_untranslated = 0
+    sum_work = 0
+    sum_total = 0
+
     for s in stats:
-        string_lengths = {k: max(string_lengths[k], len(s[k])) for k in ['name', 'branch']}
-        number_sums = {k: number_sums[k] + s[k] for k in number_keys}
+        name = max(name, len(s.name))
+        branch = max(branch, len(s.branch))
+        sum_translated = sum_translated + s.translated
+        sum_fuzzy = sum_fuzzy + s.fuzzy
+        sum_untranslated = sum_untranslated + s.untranslated
+        sum_work = sum_work + s.work
+        sum_total = sum_total + s.total
 
-    lengths = {k: len(str(number_sums[k])) for k in number_keys}
-    lengths.update(string_lengths)
-    lengths['count']=len(str(len(stats)))
-    return lengths, number_sums
+    max_len = Stat(count=len(str(len(stats))), name=name, branch=branch,
+                   translated=len(str(sum_translated)), fuzzy=len(str(sum_fuzzy)),
+                   untranslated=len(str(sum_untranslated)), work=len(str(sum_work)),
+                   total=len(str(sum_total)))
+    sums = Stat(count=len(stats), name='', branch='', translated=sum_translated,
+                fuzzy=sum_fuzzy, untranslated=sum_untranslated, work=sum_work,
+                total=sum_total)
+    return {'max_len': max_len, 'sums': sums}
 
-def display_max_lengths(lengths):
-    result = {k:max(len(titles[k]), lengths[k]) for k,v in titles.items()}
+
+def divider(s):
+    inner = '+'.join(['-' * (f + 1) for f in s]) 
+    return inner.join(['|'] * 2)
+
+
+def stat_line(max_len, s):
+    args = []
+    for i in range(len(max_len)):
+        width = max_len[i] + 1
+        method_name = 'rjust' if isinstance(s[i], int) else 'ljust'
+        part = getattr(str(s[i]), method_name)(width)[:width]
+        args.append(part)
+
+    result = '|{0}|{1}|{2}|{3}|{4}|{5}|{6}|{7}|'.format(*tuple(args))
     return result
 
-def divider(lengths):
-    
-    result = ('-' * (1 + lengths['count']) + '+' +
-              '-' * (1 + lengths['name']) + '+' +
-              '-' * (1 + lengths['branch']) + '+' +
-              '-' * (1 + lengths['work']) + '+' +
-              '-' * (1 + lengths['untranslated']) + '+' +
-              '-' * (1 + lengths['fuzzy']) + '+' +
-              '-' * (1 + lengths['translated']) + '+' +
-              '-' * (1 + lengths['total']))                  
 
+def title_line(max_len):
+    args = []
+    for i in range(len(max_len)):
+        width = max_len[i] + 1
+        method_name = titles[i][1]
+        part = getattr(titles[i][0], method_name)(width)[:width]
+        args.append(part)
+
+    result = '|{0}|{1}|{2}|{3}|{4}|{5}|{6}|{7}|'.format(*tuple(args))
     return result
 
-def ljust(length, sth):
-    s = str(sth)
-    result = (s + ' ' * length)[:length+1]
-    #print('ljust', length, sth, '>'+result+'<')
-    return result
-
-def rjust(length, sth):
-    s = str(sth)
-    result = (' ' * length + s)[-length:]
-    #print('rjust', length, sth, '>'+result+'<')
-    return result
-
-def line(lengths, stat):
-    result = (rjust(lengths['count'], stat['count']) + '|' +
-              ljust(lengths['name'], stat['name']) + '|' +
-              ljust(lengths['branch'], stat['branch']) + '|' +
-              rjust(lengths['work'], stat['work']) + '|' +
-              rjust(lengths['untranslated'], stat['untranslated']) + '|' +
-              rjust(lengths['fuzzy'], stat['fuzzy']) + '|' +
-              rjust(lengths['translated'], stat['translated']) + '|' +
-              rjust(lengths['total'], stat['total']))
-    return result
 
 def main():
@@ -125 +137 @@
     #    print('Will use {}'.format(versions[0]))
     stats = get_stats('gnome-3-18')
-    # sorts are stable
-    stats.sort(key=lambda s: s['name'])
-    stats.sort(key=lambda s: s['fuzzy'], reverse=True)
-    stats.sort(key=lambda s: s['untranslated'], reverse=True)
-    stats.sort(key=lambda s: s['work'], reverse=True)
-    print(stats)
-    lengths, sums = stats_max_lengths_sums(stats)
-    print(lengths)
-    print(divider(lengths))
-    lengths = display_max_lengths(lengths)
-    print(lengths)
-    print(divider(lengths))
-    i = 1
+    sms = stats_max_lengths_sums(stats)
+    max_len = sms['max_len']
+    sums = sms['sums']
+
+    print(max_len)
+    print(sums)
+    d = divider(max_len)
+    print(d)
+    print(title_line(max_len))
+    print(d)
     for s in stats:
-        s['count'] = i
-        i+=1
-        print(line(lengths, s))
-    print(sums)
+        print(stat_line(max_len, s))
+    print(d)
+    print(stat_line(max_len, sums))
               
 if __name__ == '__main__':