Skip to content

Instantly share code, notes, and snippets.

@joshkehn
Created December 24, 2012 06:11
Show Gist options
  • Save joshkehn/4367980 to your computer and use it in GitHub Desktop.
Save joshkehn/4367980 to your computer and use it in GitHub Desktop.
#!/usr/bin/env python
import os
import random
import time
import platform
snowflakes = {}
try:
# Windows Support
from colorama import init
init()
except ImportError:
pass
def get_terminal_size():
def ioctl_GWINSZ(fd):
try:
import fcntl
import termios
import struct
cr = struct.unpack('hh', fcntl.ioctl(fd, termios.TIOCGWINSZ,
'1234'))
except:
return None
return cr
cr = ioctl_GWINSZ(0) or ioctl_GWINSZ(1) or ioctl_GWINSZ(2)
if not cr:
try:
fd = os.open(os.ctermid(), os.O_RDONLY)
cr = ioctl_GWINSZ(fd)
os.close(fd)
except:
pass
if not cr:
try:
cr = (os.environ['LINES'], os.environ['COLUMNS'])
except:
cr = (25, 80)
return int(cr[1]), int(cr[0])
columns, rows = get_terminal_size()
def clear_screen(numlines=100):
"""Clear the console.
numlines is an optional argument used only as a fall-back.
"""
if os.name == "posix":
# Unix/Linux/MacOS/BSD/etc
os.system('clear')
elif os.name in ("nt", "dos", "ce"):
# DOS/Windows
os.system('cls')
else:
# Fallback for other operating systems.
print('\n' * rows)
def get_random_flake():
if not platform.system() == 'Windows':
try:
# python3 support
try:
cmd = unichr
except NameError:
cmd = chr
flake = cmd(random.choice(range(0x2740, 0x2749)))
return flake
except:
pass
return " *"
def get_flake_column():
return [1, get_random_flake(), random.choice(range(1, 4)), 0]
def move_flake(col):
if snowflakes[col][0]+1 == rows:
snowflakes[col] = get_flake_column()
else:
print("\033[%s;%sH " % (snowflakes[col][0], col))
snowflakes[col][3] += 1
snowflakes[col][3] %= snowflakes[col][2]
if snowflakes[col][3] == 0:
snowflakes[col][0] += 1
print("\033[%s;%sH%s" % (snowflakes[col][0], col, snowflakes[col][1]))
print("\033[1;1H")
if __name__ == "__main__":
clear_screen()
while True:
col = random.choice(range(1, int(columns)))
# its already on the screen, move it
if col in snowflakes.keys():
move_flake(col)
else:
# otherwise put it on the screen
snowflakes[col] = get_flake_column()
print("\033[%s;%sH%s" % (snowflakes[col][0], col,
snowflakes[col][1]))
# key any flakes on the screen moving
for flake in snowflakes.keys():
move_flake(flake)
time.sleep(0.025)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment