← Back to branch summary

~yasumoto7/five-a-day/icon_countdown

« back to all changes in this revision

Viewing changes to fiveaday/files.py

  • Committer: Daniel Holbach
  • Date: 2008-06-20 10:01:11 UTC
  • Revision ID: daniel.holbach@canonical.com-20080620100111-kxbixg3z148ezd2l
use lpteams to find out if user IS in a team, add convenience functions

Show diffs side-by-side

added added

removed removed

Lines of Context:
fiveaday/files.py
4
4
 
5
5
from gettext import gettext as _
6
6
 
 
7
import lpteams
 
8
 
7
9
def get_lpid():
8
10
    if not os.path.exists(os.path.expanduser("~/.5-a-day")):
9
11
        print >> sys.stderr, _("""Please type in:
141
143
    except:
142
144
        pass
143
145
 
 
146
    lp_teams = lpteams.get_teams(get_lpid())
 
147
    missing_teams = filter(lambda a: a not in lp_teams, args[1:])
144
148
    if args[0] == "--remove-team" and args[1] in teams:
145
 
        teams.remove(args[1])
 
149
        args.remove("--remove-team")
 
150
        if missing_teams:
 
151
            print >> sys.stderr, "You're not in %s, not removing." % \
 
152
                        (", ".join(missing_teams))
 
153
            for mt in missing_teams:
 
154
                args.remove(mt)
 
155
            if not args:
 
156
                return False
 
157
        for a in args:
 
158
            teams.remove(a)
 
159
 
146
160
    if args[0] == "--add-team" and args[1] not in teams:
147
 
        teams.append(args[1])
 
161
        args.remove("--add-team")
 
162
        if missing_teams:
 
163
            print >> sys.stderr, "You're not in %s, not adding." % \
 
164
                        (", ".join(missing_teams))
 
165
            for mt in missing_teams:
 
166
                args.remove(mt)
 
167
            if not args:
 
168
                return False
 
169
        for a in args:
 
170
           teams.append(a)
148
171
 
149
172
    f = open(team_file(), "w")
150
173
    f.write("\n".join(teams))
151
174
    f.close()
 
175
    return True
 
176
 
 
177
def add_to_teams(teams):
 
178
    l = ["--add-team"]
 
179
    l.extend(teams)
 
180
    modify_teams(l)
 
181
 
 
182
def remove_from_teams(teams):
 
183
    l = ["--remove-team"]
 
184
    l.extend(teams)
 
185
    modify_teams(l)
152
186
 
153
187
def show_teams():
154
188
    teams = get_teams()