vcs.maemo.org Git - maegirls/blob - trunk/src/config.py

   1 #!/usr/bin/env python
   2 # coding=UTF-8
   3 #
   4 # Copyright (C) 2010 Stefanos Harhalakis
   5 #
   6 # This file is part of maegirls.
   7 #
   8 # maegirls is free software: you can redistribute it and/or modify
   9 # it under the terms of the GNU General Public License as published by
  10 # the Free Software Foundation, either version 3 of the License, or
  11 # (at your option) any later version.
  12 #
  13 # maegirls is distributed in the hope that it will be useful,
  14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 # GNU General Public License for more details.
  17 #
  18 # You should have received a copy of the GNU General Public License
  19 # along with maegirls.  If not, see <http://www.gnu.org/licenses/>.
  20 #
  21 # $Id: 0.py 2265 2010-02-21 19:16:26Z v13 $
  22
  23 __version__ = "$Id: 0.py 2265 2010-02-21 19:16:26Z v13 $"
  24
  25 import os
  26
  27 import pickle
  28 import time
  29 import algo
  30
  31 try:
  32     home=os.environ['HOME']
  33 except:
  34     home="/home/user"
  35
  36 fn="%s/.maegirls" % (home, )
  37
  38 # Defaults for a girl
  39 defaults={
  40     'cycle':    28,
  41     'day0':     algo.today(),
  42     }
  43
  44 # Default config
  45 defaultcfg={
  46     'ver':      1,              # Configuration version
  47     'girls':    {               # List of configured girls
  48         'default':  defaults
  49         },
  50     'cur':      "default",      # Current girl
  51     }
  52
  53 def loadConfig():
  54     try:
  55         f=file(fn, "r")
  56         st=f.read()
  57         f.close()
  58         ret=pickle.loads(st)
  59     except:
  60         ret={}
  61
  62 #    print "load:", ret
  63
  64     return(ret)
  65
  66 def sanitize_before_store(cfg0):
  67     cfg={
  68         'ver':      cfg0['ver'],
  69         'cur':      str(cfg0['cur']),
  70         'girls':    {}
  71         }
  72
  73     if cfg0.has_key('girls'):
  74         for i in cfg0['girls']:
  75             cfg['girls'][str(i)]=cfg0['girls'][i]
  76
  77     return(cfg)
  78
  79 def storeConfig(cfg0):
  80     cfg0['ver']=1
  81     cfg=sanitize_before_store(cfg0)
  82 #    print "store:", cfg
  83     st=pickle.dumps(cfg)
  84     f=file(fn, "w")
  85     f.write(st)
  86     f.close()
  87
  88 def storeGirl(name, dt):
  89     # Load old
  90     cfg=loadConfig()
  91
  92     # Ensure
  93     if not cfg.has_key('girls'):
  94         cfg['girls']={}
  95
  96     # Override
  97     cfg['girls'][name]={
  98         'cycle':    dt['cycle'],
  99         'day0':     dt['day0'],
 100         }
 101
 102     storeConfig(cfg)
 103
 104 def newGirl(name):
 105     global defaults
 106
 107     storeGirl(name, defaults)
 108
 109 def loadGirls():
 110     cfg=loadConfig()
 111     try:
 112         ret=cfg['girls']
 113     except:
 114         ret={}
 115
 116     return(ret)
 117
 118 def loadGirl(name):
 119     all=loadGirls()
 120     if all.has_key(name):
 121         ret=all[name]
 122     else:
 123         ret=defaults
 124
 125     return(ret)
 126
 127 def girlExists(name):
 128     all=loadGirls()
 129     if all.has_key(name):
 130         ret=True
 131     else:
 132         ret=False
 133     return(ret)
 134
 135 def setCurrentGirl(name):
 136     cfg=loadConfig()
 137     cfg['cur']=name
 138     storeConfig(cfg)
 139
 140 def getCurrentGirl():
 141     cfg=loadConfig()
 142     try:
 143         ret=cfg['cur']
 144     except:
 145         ret='default'
 146
 147     return(ret)
 148
 149 def removeGirl(name):
 150     cfg=loadConfig()
 151     if cfg['girls'].has_key(name):
 152         cfg['girls'].pop(name)
 153         storeConfig(cfg)
 154
 155 def init():
 156     cur=getCurrentGirl()
 157     if not girlExists(cur):
 158         storeConfig(defaultcfg)
 159
 160 # vim: set ts=8 sts=4 sw=4 noet formatoptions=r ai nocindent:
 161