← Back to branch summary

~azzar1/unity/add-show-desktop-key

« back to all changes in this revision

Viewing changes to lib/common/chat.py

  • Committer: mattgiuca
  • Date: 2008-02-19 00:54:28 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:500
db: get_user and get_users now return User objects instead of dictionaries.
    This is the major part of replacing dicts with User objects, as it
    propagates upwards.

Propagated User objects up through the following modules:
    listusers.py, dispatch.login, authenticate, userservice, forumutil
All of these now treat users as an object rather than a dict.

To save on the size of the changes so far, login still individually copies
fields over to the session (so the session does not yet store a user object;
that is the second part of this refactor).

WOO!! Revision 500 :)

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/common/chat.py
 
1
# IVLE - Informatics Virtual Learning Environment
 
2
# Copyright (C) 2007-2008 The University of Melbourne
 
3
#
 
4
# This program is free software; you can redistribute it and/or modify
 
5
# it under the terms of the GNU General Public License as published by
 
6
# the Free Software Foundation; either version 2 of the License, or
 
7
# (at your option) any later version.
 
8
#
 
9
# This program is distributed in the hope that it will be useful,
 
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
# GNU General Public License for more details.
 
13
#
 
14
# You should have received a copy of the GNU General Public License
 
15
# along with this program; if not, write to the Free Software
 
16
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
17
 
 
18
# Module: Chat
 
19
# Author: Thomas Conway
 
20
# Date:   5/2/2008
 
21
 
 
22
import cjson
 
23
import cStringIO
 
24
import md5
 
25
import os
 
26
import socket
 
27
 
 
28
def start_server(port, magic, daemon_mode, handler, initializer = None):
 
29
    # Attempt to open the socket.
 
30
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 
31
    s.bind(('', port))
 
32
    s.listen(1)
 
33
 
 
34
    # Excellent! It worked. Let's turn ourself into a daemon,
 
35
    # then get on with the job of being a python interpreter.
 
36
    if daemon_mode:
 
37
        if os.fork():   # launch child and...
 
38
            os._exit(0) # kill off parent
 
39
        os.setsid()
 
40
        if os.fork():   # launch child and...
 
41
            os._exit(0) # kill off parent again.
 
42
        os.umask(077)
 
43
 
 
44
    if initializer:
 
45
        initializer()
 
46
 
 
47
    while True:
 
48
        (conn, addr) = s.accept()
 
49
        try:
 
50
            # Grab the input
 
51
            buf = cStringIO.StringIO()
 
52
            blk = conn.recv(1024)
 
53
            while blk:
 
54
                buf.write(blk)
 
55
                try:
 
56
                    blk = conn.recv(1024, socket.MSG_DONTWAIT)
 
57
                except:
 
58
                    # Exception thrown if it WOULD block (but we
 
59
                    # told it not to wait) - ie. we are done
 
60
                    blk = None
 
61
            inp = buf.getvalue()
 
62
            env = cjson.decode(inp)
 
63
            
 
64
            # Check that the message is 
 
65
            digest = md5.new(env['content'] + magic).digest().encode('hex')
 
66
            if env['digest'] != digest:
 
67
                conn.close()
 
68
                continue
 
69
 
 
70
            content = cjson.decode(env['content'])
 
71
 
 
72
            response = handler(content)
 
73
 
 
74
            conn.sendall(cjson.encode(response))
 
75
 
 
76
            conn.close()
 
77
        except Exception, e:
 
78
            conn.sendall(cjson.encode(repr(e)))
 
79
            conn.close()
 
80
 
 
81
 
 
82
def chat(host, port, msg, magic, decode = True):
 
83
    sok = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 
84
    sok.connect((host, port))
 
85
    content = cjson.encode(msg)
 
86
    digest = md5.new(content + magic).digest().encode("hex")
 
87
    env = {'digest':digest,'content':content}
 
88
    sok.send(cjson.encode(env))
 
89
 
 
90
    buf = cStringIO.StringIO()
 
91
    blk = sok.recv(1024)
 
92
    while blk:
 
93
        buf.write(blk)
 
94
        try:
 
95
            blk = conn.recv(1024, socket.MSG_DONTWAIT)
 
96
        except:
 
97
            # Exception thrown if it WOULD block (but we
 
98
            # told it not to wait) - ie. we are done
 
99
            blk = None
 
100
    inp = buf.getvalue()
 
101
    sok.close()
 
102
 
 
103
    if decode:
 
104
        return cjson.decode(inp)
 
105
    else:
 
106
        return inp
 
107