mirror of
https://github.com/tildeclub/ttbp.git
synced 2026-06-15 17:50:18 +00:00
dccb1570d8
holy crap! here i am, about a decade and a half after my first Formal Instruction in programming, and i suddenly understand code documentation. part of this is that sometimes i stare at functions forgetting why they're there, or start writing a function with the distinct feeling that i'm typing code i've already typed before, and realize that maybe there's a better way. so i skimmed code from other people that i've used in my own repos, and lifted the general gist of their commenting style while doing things that feel right to me. i still don't know exactly how i like things, but i'm learning. this is the best way i learn things. then. then! i learned that i can just pydoc any of my modules and pydoc will generate literally the same thing that i read when i pydoc other module! what. WHAT. this is amazing. i feel like a real person. i understand where those docs come from now, and how to make them myself. i'm learning so much. why does this feel so amazing. all of this is in a commit message that i'm going to fire off into the sun but i just need to put this out there because this feels important.
128 lines
3.0 KiB
Python
128 lines
3.0 KiB
Python
#!/usr/bin/python
|
|
|
|
'''
|
|
util.py: frequently used terminal and text processing utilities
|
|
copyright (c) 2016 ~endorphant (endorphant@tilde.town)
|
|
|
|
GNU GPL BOILERPLATE:
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
'''
|
|
|
|
import inflect
|
|
import time
|
|
import random
|
|
import colorama
|
|
|
|
colorama.init()
|
|
|
|
textcolors = [ colorama.Fore.RED, colorama.Fore.GREEN, colorama.Fore.YELLOW, colorama.Fore.BLUE, colorama.Fore.MAGENTA, colorama.Fore.WHITE, colorama.Fore.CYAN]
|
|
|
|
lastcolor = colorama.Fore.RESET
|
|
|
|
p = inflect.engine()
|
|
|
|
def set_rainbow():
|
|
'''
|
|
prints a random terminal color code
|
|
'''
|
|
|
|
global lastcolor
|
|
|
|
color = lastcolor
|
|
while color == lastcolor:
|
|
color = random.choice(textcolors)
|
|
|
|
lastcolor = color
|
|
|
|
print(color)
|
|
|
|
def reset_color():
|
|
'''
|
|
prints terminal color code reset
|
|
'''
|
|
|
|
print(colorama.Fore.RESET)
|
|
|
|
def attach_rainbow():
|
|
'''
|
|
returns a random terminal color code, presumably to be 'attached' to a string
|
|
'''
|
|
|
|
global lastcolor
|
|
|
|
color = lastcolor
|
|
while color == lastcolor:
|
|
color = random.choice(textcolors)
|
|
|
|
lastcolor = color
|
|
return color
|
|
|
|
def attach_reset():
|
|
'''
|
|
returns terminal color code reset, presumably to be 'attached' to a string
|
|
'''
|
|
|
|
return colorama.Style.RESET_ALL
|
|
|
|
def hilight(text):
|
|
'''
|
|
takes a string and highlights it on return
|
|
'''
|
|
|
|
return colorama.Style.BRIGHT+text+colorama.Style.NORMAL
|
|
|
|
def pretty_time(time):
|
|
'''
|
|
human-friendly time formatter
|
|
|
|
takes an integer number of seconds and returns a phrase that describes it,
|
|
using the largest possible figure, rounded down (ie, time=604 returns '10
|
|
minutes', not '10 minutes, 4 seconds' or '604 seconds')
|
|
'''
|
|
|
|
m, s = divmod(time, 60)
|
|
if m > 0:
|
|
h, m = divmod(m, 60)
|
|
if h > 0:
|
|
d, h = divmod(h, 24)
|
|
if d > 0:
|
|
w, d = divmod(d, 7)
|
|
if w > 0:
|
|
mo, w = divmod(w, 4)
|
|
if mo > 0:
|
|
return p.no("month", mo)
|
|
else:
|
|
return p.no("week", w)
|
|
else:
|
|
return p.no("day", d)
|
|
else:
|
|
return p.no("hour", h)
|
|
else:
|
|
return p.no("minute", m)
|
|
else:
|
|
return p.no("second", s)
|
|
|
|
def genID(digits=5):
|
|
'''
|
|
returns a string-friendly string of digits, which can start with 0
|
|
'''
|
|
|
|
id = ""
|
|
x = 0
|
|
while x < digits:
|
|
id += str(random.randint(0,9))
|
|
x += 1
|
|
|
|
return id
|