rain.py

#!/usr/bin/env python
#
# $Id$
#
# somebody should probably check the randrange()s...

importcurses
fromrandomimportrandrange

defnext_j(j):
ifj==0:
j=4
else:
j-=1

ifcurses.has_colors():
z=randrange(0, 3)
color=curses.color_pair(z)
ifz:
color=color|curses.A_BOLD
stdscr.attrset(color)

returnj

defmain(win):
# we know that the first argument from curses.wrapper() is stdscr.
# Initialize it globally for convenience.
globalstdscr
stdscr=win

ifcurses.has_colors():
bg=curses.COLOR_BLACK
curses.init_pair(1, curses.COLOR_BLUE, bg)
curses.init_pair(2, curses.COLOR_CYAN, bg)

curses.nl()
curses.noecho()
# XXX curs_set() always returns ERR
# curses.curs_set(0)
stdscr.timeout(0)

c=curses.COLS-4
r=curses.LINES-4
xpos= [0] *c
ypos= [0] *r
forjinrange(4, -1, -1):
xpos[j] =randrange(0, c) +2
ypos[j] =randrange(0, r) +2

j=0
whileTrue:
x=randrange(0, c) +2
y=randrange(0, r) +2

stdscr.addch(y, x, ord('.'))

stdscr.addch(ypos[j], xpos[j], ord('o'))

j=next_j(j)
stdscr.addch(ypos[j], xpos[j], ord('O'))

j=next_j(j)
stdscr.addch( ypos[j] -1, xpos[j], ord('-'))
stdscr.addstr(ypos[j], xpos[j] -1, "|.|")
stdscr.addch( ypos[j] +1, xpos[j], ord('-'))

j=next_j(j)
stdscr.addch( ypos[j] -2, xpos[j], ord('-'))
stdscr.addstr(ypos[j] -1, xpos[j] -1, "/ \\")
stdscr.addstr(ypos[j], xpos[j] -2, "| O |")
stdscr.addstr(ypos[j] +1, xpos[j] -1, "\\ /")
stdscr.addch( ypos[j] +2, xpos[j], ord('-'))

j=next_j(j)
stdscr.addch( ypos[j] -2, xpos[j], ord(' '))
stdscr.addstr(ypos[j] -1, xpos[j] -1, " ")
stdscr.addstr(ypos[j], xpos[j] -2, " ")
stdscr.addstr(ypos[j] +1, xpos[j] -1, " ")
stdscr.addch( ypos[j] +2, xpos[j], ord(' '))

xpos[j] =x
ypos[j] =y

ch=stdscr.getch()
ifch==ord('q') orch==ord('Q'):
return
elifch==ord('s'):
stdscr.nodelay(0)
elifch==ord(' '):
stdscr.nodelay(1)

curses.napms(50)

curses.wrapper(main)