-
Notifications
You must be signed in to change notification settings - Fork 0
/
finput.py
70 lines (59 loc) · 2.16 KB
/
finput.py
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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Some helpers for file input.
import os
def humansize(s):
'''
Return a string representing a human readable representation of a
filesize
'''
_abbrevs = [(1<<50,'PiB'), (1<<40,'TiB'), (1<<30,'GiB'),
(1<<20,'MiB'), (1<<10,'KiB'), (1, 'B')]
size = os.path.getsize(s)
for factor, suffix in _abbrevs:
if size >= factor: break
if suffix == 'B': return '%d %s' % (size/float(factor), suffix)
return '%0.2f %s' % (size/float(factor), suffix)
def sortedginfo(ss):
def ginfo(s):
nfo = "?"
if os.path.isdir(s):
nfo = "<DIR>"
else:
try: nfo = humansize(s)
except: pass
return (nfo, os.path.split(s)[-1])
li = [ginfo(x) for x in ss]
dirs = list(filter(lambda x: x[0] == "<DIR>", li))
dirs.sort(key=lambda x: x[1].upper())
fils = list(filter(lambda x: x[0] != "<DIR>", li))
fils.sort(key=lambda x: x[1].upper())
return ["%12s %s" % x for x in dirs+fils]
def tryget(s):
"""
Try and get contents of s. Return a tuple of: (content, message, success)
- `content` is a file-descriptor, if it's a normal, readable file or None.
- `message` may contain:
- the directory-content if s is a dir
- an informative message about the file being read if s is a file.
- Some errormessage if anything goes wrong
- `success` will be False if anything fails (file/dir not found/not readable)
"""
if not os.path.exists(s):
return (None, s+ " does not exist.", False)
if os.path.isfile(s):
try:
f = open(s, "rb")
t = os.path.getsize(s)
return(f, "Opened file %s, %d bytes" % (s, t), True)
except e:
return (None, str(e), False)
if os.path.isdir(s):
try:
li = sortedginfo(os.path.join(s,x) for x in os.listdir(s))
except Exception as e:
return (None, str(e), False)
res = "\n\n----- DIRECTORY %s -----\n\n" % s
res += "\n".join(li)
return (None, res, True)
return(None, s+" is not a regular file (or dir)", False)