File: MT_users2csv.py

package info (click to toggle)
childsplay-plugins 0.84-1
  • links: PTS
  • area: main
  • in suites: etch, etch-m68k
  • size: 4,804 kB
  • ctags: 351
  • sloc: python: 3,066; sh: 129; makefile: 6
file content (95 lines) | stat: -rw-r--r-- 3,067 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Copyright (c) 2006 Stas Zykiewicz <stas.zytkiewicz@gmail.com>
#
#           MT_users2csv.py
# This program is free software; you can redistribute it and/or
# modify it under the terms of version 2 of the GNU General Public License
# as published by the Free Software Foundation.  A copy of this license should
# be included in the file GPL-2.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.



"""This is a simple script that turn the values from the multiTable plugin
into a CSV list suitable for importing in a database.
From there the data can printed or whatever.
This is mainly intended for teachers who want to use multiTable as a tool to
track the childrens progress.

Usage: MT_users2csv.py mt_users
The path is optional.
The csv is saved into the current directory when no path is given.
"""

import os,sys,pickle,time

print __doc__

USERSFILE = os.path.join(os.path.expanduser('~/.childsplay'),'multiTables','mt_users')
DESTFILE = os.path.join(os.path.expanduser('~'),'mt_users-%s.csv' % 
                                time.strftime("%d_%m_%y-%X",time.localtime()))

if not os.path.exists(USERSFILE):
    print "Can't find the users file in %s" % USERSFILE
    
def _fetch_sheet():
    """ Load the users file, makes a default sheet when the load fails."""
    try:
        f = open(USERSFILE,'r')
        sheet = pickle.load(f)
        f.close()
    except (IOError,EOFError),info:
        print info
        print >> sys.stderr,'Error in fetching users file'
        print >> sys.stderr,info
        sys.exit(1)
    return sheet

if __name__ == '__main__':
    csv = []
    sheet = _fetch_sheet()
    for user,dic in sheet.items():
        # split the dic in tables and others.
        # the key for a table is always a str(int) so we test for int(key)
        # when it fails it belongs to other, otherwise it will be a table
        others = []
        tables = []
        for k,v in dic.items():
            try:
                int(k)
            except ValueError:
                others.append((k,v))
            else:
                tables.append(('%2s' % k,str(v[1])))
        tables.sort()
        table_times = []
        for item in tables:
            table_times.append(item[1])
        others.sort()
        others_vals = []
        for item in others:
            others_vals.append(str(item[1]))
        csv.append(['"%s"' % user]+table_times+others_vals)
    
    try:
        f = open(DESTFILE,'w')
    except IOError,info:
        print info
        sys.exit(1)
    for line in csv:
        #print line
        l = ','.join(line)
        f.write(l+'\n')
    f.close()