Example #1
0
 def addMessage(self, nick, network_name, channel_name, message):
     user = self.addUser(nick)
     network = self.addNetwork(network_name)
     channel = self.addChannel(channel_name)
     message = Message(user, network, channel, message)
     db.add(message)
     if not self.commit():
         raise ModelException("Problem committing message to %s by %s \n\t %s" % (network, user, message))
     print "Committing add of %s" % message
     return user
Example #2
0
 def addUser(self, nick):
     user = User.query.filter(User.nick == nick).first()
     if user is not None:
         return user
     user = User(nick)
     db.add(user)
     if not self.commit():
         raise ModelException("Problem committing user with nick " + nick)
     print "Committing add of %s" % nick
     return user
Example #3
0
 def addChannel(self, name):
     channel = Channel.query.filter(Channel.name == name).first()
     if channel is not None:
         return channel
     channel = Channel(name)
     db.add(channel)
     if not self.commit():
         raise ModelException("Problem committing channel " + name)
     print "Committing add of %s" % name
     return channel
Example #4
0
 def addNetwork(self, network_name):
     network = Network.query.filter(Network.name == network_name).first()
     if network is not None:
         return network
     network = Network(network_name)
     db.add(network)
     if not self.commit():
         raise ModelException("Problem committing network " + network_name)
     print "Committing add of %s" % network_name
     return network
Example #5
0
 def addURL(self, message_id, url):
     message = Message.query.filter(Message.mid == int(message_id)).first()
     if message is None:
         print "Unable to find message with ID %d" % int(message_id)
         return None
     url = Url(message_id, url)
     db.add(url)
     if not self.commit():
         raise ModelException("Problem committing url %s for message_id %s" % (url, message_id))
     print "Committing add of %s" % url
     return url
Example #6
0
    def addBot(self, nick, network_name):
        if network_name is None or len(network_name.strip()) == 0:
            raise ModelException("Network required when adding a bot.")

        network = self.addNetwork(network_name)

        bot = Bot.query.filter(Bot.nick == nick).filter(Bot.network == network).first()
        if bot is not None:
            return bot

        bot = Bot(nick, network_name)
        db.add(bot)
        if not self.commit():
            raise ModelException("Error during commit of Bot " + nick + " for network " + network_name)
        print "Committing add of %s to %s" % (nick, network_name)
        return bot
Example #7
0
    def addServer(self, network_name, address, port=None, SSL=False):
        server = Server.query.filter(Server.address == address).first()
        if server is not None:
            return server

        #Make sure the network exists
        self.addNetwork(network_name)

        if port is None or len(port.strip()) == 0:
            port = None
        else:
            port = int(port)
        server = Server(network_name, address, port, SSL)
        db.add(server)
        if self.commit():
            #TODO work out how to print out the server here using the toString shit
            raise ModelException("Problem committing server " + str(server))
        print "Committing add of %s to %s" % (address, network_name)
        return server