Skip to content

Instantly share code, notes, and snippets.

@pavellishin
Forked from sontek/snowjob.py
Created December 22, 2011 15:45
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save pavellishin/1510737 to your computer and use it in GitHub Desktop.
Save pavellishin/1510737 to your computer and use it in GitHub Desktop.
Make your terminal snow with python
#!/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 move_flake(col):
if snowflakes[col][0]+1 == rows:
snowflakes[col] = [1, get_random_flake()]
else:
print( ("\033[%s;%sH " % (snowflakes[col][0], col)).encode("utf8") )
snowflakes[col][0] += 1
print( ("\033[%s;%sH%s" % (snowflakes[col][0], col, snowflakes[col][1])).encode("utf8") )
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
flake = get_random_flake()
snowflakes[col] = [1, flake]
print( ("\033[%s;%sH%s" % (snowflakes[col][0],
col,
snowflakes[col][1])).encode("utf8") )
# key any flakes on the screen moving
for flake in snowflakes.keys():
move_flake(flake)
time.sleep(0.1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment