couchpytato.py

   1 #!/usr/bin/env python
   2
   3 try:
   4     # trying to import 'psyco' which precompiles
   5     # the program in order to make it faster
   6     # [http://psyco.sourceforge.net/]
   7     import psyco
   8     psyco.full()
   9 except ImportError:
  10     pass
  11
  12 import pygame, os, time, sys, configobj
  13 from pygame.locals import *
  14 sys.path.append(os.path.join(sys.path[0], "data"))
  15
  16 class couchpytato:
  17     def __init__(self):
  18         cf = configobj.ConfigObj(os.path.join(sys.path[0], 'couchpytato.cfg'))
  19         if cf['sound'] == 'pysonic':
  20             import pySonic
  21             w = pySonic.World()
  22         pygame.display.init()
  23         pygame.font.init()
  24         import cfg, playlist
  25         resx = int(cfg.th['resolution'][0])
  26         resy = int(cfg.th['resolution'][1])
  27         pygame.display.set_caption('couchpytato')
  28         pygame.display.set_icon(pygame.image.load(os.path.join(sys.path[0], 'icon.png')))
  29         if cfg.cf['fullscreen'] == '1':
  30             screen = pygame.display.set_mode((resx, resy),pygame.FULLSCREEN)
  31         else:
  32             screen = pygame.display.set_mode((resx, resy))
  33         pygame.mouse.set_visible(False)
  34         pygame.event.set_blocked(MOUSEMOTION)
  35         pygame.key.set_repeat(200, 80)
  36         clock = pygame.time.Clock()
  37
  38         joystick = None
  39         if pygame.joystick.get_init():
  40             if pygame.joystick.get_count() > 0:
  41                 joystick = pygame.joystick.Joystick(0)
  42                 joystick.init()
  43
  44         mod = __import__('mod_mainmenu')
  45         pytato = mod.Module(screen)
  46         playlist.PLAYLIST.load()
  47
  48         while 1:
  49             pytato.runloop()
  50             clock.tick(int(cfg.cf['fps']))
  51             for event in pygame.event.get((KEYDOWN, KEYUP, QUIT)):
  52                 if event.type == QUIT:
  53                     self.quit()
  54                 if pytato.loadmodule:
  55                     try:
  56                         mod = __import__(pytato.loadmodule)
  57                     except:
  58                         mod = __import__(cfg.cf['modules'][pytato.loadmodule]['modfilename'])
  59                     pytato = mod.Module(screen)
  60                     pytato.loadmodule = None
  61                 pytato.keyhandler(event)
  62                 pytato.generalkeyhandler(event)
  63             if joystick: pytato.joyhandler(joystick)
  64
  65     def quit(self):
  66         import playlist
  67         playlist.PLAYLIST.save()
  68         pygame.quit()
  69         sys.exit()
  70
  71 couchpytato()