1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
--- temp_summary.py.orig 2003-05-06 11:53:14 UTC
+++ temp_summary.py
@@ -2,11 +2,9 @@
from ed2k_metutils import *
import os
import stat
+import unicodedata
-# I'm really surprised there's no easy way to get the terminal
-# width in python... :-/ I can't do external invocations to
-# stty, etc, because they might not be there on windows...
-WIDTH = 80;
+WIDTH = getattr(os.get_terminal_size(), 'columns', 80)
if __name__ == "__main__":
# Here's an example to cut and keep.
@@ -15,13 +13,10 @@ if __name__ == "__main__":
# see how much data I actually got from night to night.
if len( sys.argv ) < 2:
- print "invocation: %s < <x.part.met> [x.part.met ...] | <temp_dir> >" % sys.argv[ 0 ];
- print
- print "This program will show the amount downloaded vs. the total size "
- print "for the .part.met files listed on the command line."
- print
- print "This program assumes an 80 column display. You can tweak this "
- print "by editing the script. Change the 'WIDTH' value."
+ print("invocation: %s < <x.part.met> [x.part.met ...] | <temp_dir> >" % sys.argv[0])
+ print()
+ print("This program will show the amount downloaded vs. the total size")
+ print("for the .part.met files listed on the command line.")
sys.exit( -1 );
total_size = total_down = 0;
@@ -34,7 +29,7 @@ if __name__ == "__main__":
for met_file in mets:
- fh = open( met_file, "r" );
+ fh = open(met_file, "rb")
data = fh.read();
fh.close();
@@ -43,7 +38,7 @@ if __name__ == "__main__":
# We're interested in the name, the total size, and some kind of... anti-gapping.
size = met_data.FindTags( TAG_HANDLE_FILESIZE )[ 0 ].value;
- name = met_data.FindTags( TAG_HANDLE_FILENAME )[ 0 ].value;
+ name = met_data.FindTags(TAG_HANDLE_FILENAME)[0].value.decode()
# Set the total downloaded to the file size.
down = size;
@@ -71,19 +66,42 @@ if __name__ == "__main__":
bar = "#" * ( WIDTH - 2 );
for gap in gaps:
gap_start, gap_end = gaps[ gap ];
- char_gap_start = gap_start / bytes_per_char;
- char_gap_end = gap_end / bytes_per_char;
+ char_gap_start = int(gap_start / bytes_per_char)
+ char_gap_end = int(gap_end / bytes_per_char)
bar = bar[ : char_gap_start ] + " " * ( char_gap_end - char_gap_start ) + bar[ char_gap_end : ];
+
+ # Account for CJK characters occupy two fixed-width spaces.
+ def char_width(c: str) -> int:
+ if not c.isprintable(): return 0
+ return 2 if unicodedata.category(c) == 'Lo' else 1
+
+ def visible_len(s: str) -> int:
+ return sum(char_width(c) for c in s)
+
+ # Truncate string to specified limit. If truncation happens
+ # on double-width character (like it would have to be cut in
+ # half), append an extra space for nicer alignment.
+ def visible_substr_padded(s: str, l: int) -> str:
+ vislen = 0
+ cut_here = 0
+ padding = ''
+ for c in s:
+ vislen += char_width(c)
+ if vislen <= l: cut_here += 1
+ if vislen == l: break
+ if vislen > l: padding = ' '; break
+ return s[:cut_here] + padding
# Print out our summary. Limit the filenames.
- sizestring = " - %s - %iK of %iK" % ( met_file.split( "/" )[ -1 ], down / 1024, size / 1024 );
+ sizestring = " - %s - %.2fK of %.2fK" % (met_file.split("/")[-1], down / 1024, size / 1024)
max_name_size = WIDTH - len( sizestring );
- if len( name ) < max_name_size:
- name += " " * ( max_name_size - len( name ) );
+ vislen = visible_len(name)
+ if vislen < max_name_size:
+ name += " " * (max_name_size - vislen)
else:
- name = name[ : max_name_size ];
- print "%s%s" % ( name, sizestring );
- print "[%s]" % bar;
- print
+ name = visible_substr_padded(name, max_name_size)
+ print("%s%s" % (name, sizestring))
+ print("[%s]" % bar)
+ print()
del( met_data );
- print "Totals: %sK of %sK" % ( total_down / 1024, total_size / 1024 );
+ print("Totals: %.2fK of %.2fK" % (total_down / 1024, total_size / 1024))
|
