""" Models for the comsystem. The Commsystem is intended to be used by Players (thematic IC communication is probably best handled by custom commands instead). The comm system could take the form of channels, but can also be adopted for storing tells or in-game mail. The comsystem's main component is the Message (Msg), which carries the actual information between two parties. Msgs are stored in the database and usually not deleted. A Msg always have one sender (a user), but can have any number targets, both users and channels. Channels are central objects that act as targets for Msgs. Players can connect to channels by use of a ChannelConnect object (this object is necessary to easily be able to delete connections on the fly). """ from datetime import datetime from django.conf import settings from django.db import models from src.typeclasses.models import TypedObject, TagHandler, AttributeHandler, AliasHandler from src.utils.idmapper.models import SharedMemoryModel from src.comms import managers from src.comms.managers import identify_object from src.locks.lockhandler import LockHandler from src.utils import logger from src.utils.utils import is_iter, to_str, crop, make_iter __all__ = ("Msg", "TempMsg", "ChannelDB", "PlayerChannelConnection", "ExternalChannelConnection") _GA = object.__getattribute__ _SA = object.__setattr__ _DA = object.__delattr__ #------------------------------------------------------------ # # Msg # #------------------------------------------------------------ class Msg(SharedMemoryModel): """ A single message. This model describes all ooc messages sent in-game, both to channels and between players. The Msg class defines the following properties: sender - sender of message receivers - list of target objects for message channels - list of channels message was sent to message - the text being sent date_sent - time message was sent hide_from_sender - bool if message should be hidden from sender hide_from_receivers - list of receiver objects to hide message from hide_from_channels - list of channels objects to hide message from permissions - perm strings """ # # Msg database model setup # # # These databse fields are all set using their corresponding properties, # named same as the field, but withtout the db_* prefix. # Sender is either a player, an object or an external sender, like an IRC channel # normally there is only one, but if co-modification of a message is allowed, there # may be more than one "author" db_sender_players = models.ManyToManyField("players.PlayerDB", related_name='sender_player_set', null=True, verbose_name='sender(player)', db_index=True) db_sender_objects = models.ManyToManyField("objects.ObjectDB", related_name='sender_object_set', null=True, verbose_name='sender(object)', db_index=True) db_sender_external = models.CharField('external sender', max_length=255, null=True, db_index=True, help_text="identifier for external sender, for example a sender over an IRC connection (i.e. someone who doesn't have an exixtence in-game).") # The destination objects of this message. Stored as a # comma-separated string of object dbrefs. Can be defined along # with channels below. db_receivers_players = models.ManyToManyField('players.PlayerDB', related_name='receiver_player_set', null=True, help_text="player receivers") db_receivers_objects = models.ManyToManyField('objects.ObjectDB', related_name='receiver_object_set', null=True, help_text="object receivers") db_receivers_channels = models.ManyToManyField("ChannelDB", related_name='channel_set', null=True, help_text="channel recievers") # header could be used for meta-info about the message if your system needs it, or as a separate # store for the mail subject line maybe. db_header = models.TextField('header', null=True, blank=True) # the message body itself db_message = models.TextField('messsage') # send date db_date_sent = models.DateTimeField('date sent', editable=False, auto_now_add=True, db_index=True) # lock storage db_lock_storage = models.TextField('locks', blank=True, help_text='access locks on this message.') # these can be used to filter/hide a given message from supplied objects/players/channels db_hide_from_players = models.ManyToManyField("players.PlayerDB", related_name='hide_from_players_set', null=True) db_hide_from_objects = models.ManyToManyField("objects.ObjectDB", related_name='hide_from_objects_set', null=True) db_hide_from_channels = models.ManyToManyField("ChannelDB", related_name='hide_from_channels_set', null=True) # Database manager objects = managers.MsgManager() def __init__(self, *args, **kwargs): SharedMemoryModel.__init__(self, *args, **kwargs) self.locks = LockHandler(self) self.extra_senders = [] class Meta: "Define Django meta options" verbose_name = "Message" # Wrapper properties to easily set database fields. These are # @property decorators that allows to access these fields using # normal python operations (without having to remember to save() # etc). So e.g. a property 'attr' has a get/set/del decorator # defined that allows the user to do self.attr = value, # value = self.attr and del self.attr respectively (where self # is the object in question). # sender property (wraps db_sender_*) #@property def __senders_get(self): "Getter. Allows for value = self.sender" return [hasattr(o, "typeclass") and o.typeclass or o for o in list(self.db_sender_players.all()) + list(self.db_sender_objects.all()) + self.extra_senders] #@sender.setter def __senders_set(self, value): "Setter. Allows for self.sender = value" for val in (v for v in make_iter(value) if v): obj, typ = identify_object(val) if typ == 'player': self.db_sender_players.add(obj) elif typ == 'object': self.db_sender_objects.add(obj) elif typ == 'external': self.db_sender_external = "1" self.extra_senders.append(obj) print "I ran!" elif isinstance(typ, basestring): self.db_sender_external = obj elif not obj: return else: raise ValueError(obj) self.save() #@sender.deleter def __senders_del(self): "Deleter. Clears all senders" self.db_sender_players.clear() self.db_sender_objects.clear() self.db_sender_external = "" self.extra_senders = [] self.save() senders = property(__senders_get, __senders_set, __senders_del) def remove_sender(self, value): "Remove a single sender or a list of senders" for val in make_iter(value): obj, typ = identify_object(val) if typ == 'player': self.db_sender_players.remove(obj) elif typ == 'object': self.db_sender_objects.remove(obj) elif typ == 'external': self.extra_senders = [receiver for receiver in self.extra_senders if receiver != obj] elif isinstance(obj, basestring): self.db_sender_external = obj else: raise ValueError(obj) self.save() # receivers property #@property def __receivers_get(self): "Getter. Allows for value = self.receivers. Returns three lists of receivers: players, objects and channels." return [hasattr(o, "typeclass") and o.typeclass or o for o in list(self.db_receivers_players.all()) + list(self.db_receivers_objects.all())] #@receivers.setter def __receivers_set(self, value): "Setter. Allows for self.receivers = value. This appends a new receiver to the message." for val in (v for v in make_iter(value) if v): obj, typ = identify_object(val) if typ == 'player': self.db_receivers_players.add(obj) elif typ == 'object': self.db_receivers_objects.add(obj) elif not obj: return else: raise ValueError self.save() #@receivers.deleter def __receivers_del(self): "Deleter. Clears all receivers" self.db_receivers_players.clear() self.db_receivers_objects.clear() self.extra_senders = [] self.save() receivers = property(__receivers_get, __receivers_set, __receivers_del) def remove_receiver(self, obj): "Remove a single recevier" obj, typ = identify_object(obj) if typ == 'player': self.db_receivers_players.remove(obj) elif typ == 'object': self.db_receivers_objects.remove(obj) else: raise ValueError self.save() # channels property #@property def __channels_get(self): "Getter. Allows for value = self.channels. Returns a list of channels." return self.db_receivers_channels.all() #@channels.setter def __channels_set(self, value): "Setter. Allows for self.channels = value. Requires a channel to be added." for val in (v.dbobj for v in make_iter(value) if v): self.db_receivers_channels.add(val) #@channels.deleter def __channels_del(self): "Deleter. Allows for del self.channels" self.db_receivers_channels.clear() self.save() channels = property(__channels_get, __channels_set, __channels_del) def __hide_from_get(self): "Getter. Allows for value = self.hide_from. Returns 3 lists of players, objects and channels" return self.db_hide_from_players.all(), self.db_hide_from_objects.all(), self.db_hide_from_channels.all() #@hide_from_sender.setter def __hide_from_set(self, value): "Setter. Allows for self.hide_from = value. Will append to hiders" obj, typ = identify_object(value) if typ == "player": self.db_hide_from_players.add(obj) elif typ == "object": self.db_hide_from_objects.add(obj) elif typ == "channel": self.db_hide_from_channels.add(obj) else: raise ValueError self.save() #@hide_from_sender.deleter def __hide_from_del(self): "Deleter. Allows for del self.hide_from_senders" self.db_hide_from_players.clear() self.db_hide_from_objects.clear() self.db_hide_from_channels.clear() self.save() hide_from = property(__hide_from_get, __hide_from_set, __hide_from_del) # # Msg class methods # def __str__(self): "This handles what is shown when e.g. printing the message" senders = ",".join(obj.key for obj in self.senders) receivers = ",".join(["[%s]" % obj.key for obj in self.channels] + [obj.key for obj in self.receivers]) return "%s->%s: %s" % (senders, receivers, crop(self.message, width=40)) #------------------------------------------------------------ # # TempMsg # #------------------------------------------------------------ class TempMsg(object): """ This is a non-persistent object for sending temporary messages that will not be stored. It mimics the "real" Msg object, but don't require sender to be given. """ def __init__(self, senders=None, receivers=None, channels=None, message="", header="", type="", lockstring="", hide_from=None): self.senders = senders and make_iter(senders) or [] self.receivers = receivers and make_iter(receivers) or [] self.channels = channels and make_iter(channels) or [] self.type = type self.header = header self.message = message self.lock_storage = lockstring self.locks = LockHandler(self) self.hide_from = hide_from and make_iter(hide_from) or [] self.date_sent = datetime.now() def __str__(self): "This handles what is shown when e.g. printing the message" senders = ",".join(obj.key for obj in self.senders) receivers = ",".join(["[%s]" % obj.key for obj in self.channels] + [obj.key for obj in self.receivers]) return "%s->%s: %s" % (senders, receivers, crop(self.message, width=40)) def remove_sender(self, obj): "Remove a sender or a list of senders" for o in make_iter(obj): try: self.senders.remove(o) except ValueError: pass # nothing to remove def remove_receiver(self, obj): "Remove a sender or a list of senders" for o in make_iter(obj): try: self.senders.remove(o) except ValueError: pass # nothing to remove def access(self, accessing_obj, access_type='read', default=False): "checks lock access" return self.locks.check(accessing_obj, access_type=access_type, default=default) #------------------------------------------------------------ # # Channel # #------------------------------------------------------------ class ChannelDB(TypedObject): """ This is the basis of a comm channel, only implementing the very basics of distributing messages. The Channel class defines the following properties: key - main name for channel desc - optional description of channel aliases - alternative names for the channel keep_log - bool if the channel should remember messages permissions - perm strings """ # Database manager objects = managers.ChannelManager() _typeclass_paths = settings.COMM_TYPECLASS_PATHS _default_typeclass_path = settings.BASE_COMM_TYPECLASS or "src.comms.comms.Comm" def __init__(self, *args, **kwargs): TypedObject.__init__(self, *args, **kwargs) _SA(self, "tags", TagHandler(self, category_prefix="comm_")) _SA(self, "aliases", AliasHandler(self, category_prefix="comm_")) _SA(self, "attributes", AttributeHandler(self)) class Meta: "Define Django meta options" verbose_name = "Channel" verbose_name_plural = "Channels" # # Channel class methods # def __str__(self): return "Channel '%s' (%s)" % (self.key, self.typeclass.db.desc) def has_connection(self, player): """ Checks so this player is actually listening to this channel. """ # also handle object.player calls player, typ = identify_object(player) if typ == 'object': player = player.player player, typ = identify_object(player) if player and not typ == "player": logger.log_errmsg("Channel.has_connection received object of type '%s'. It only accepts players/characters." % typ) return # do the check return PlayerChannelConnection.objects.has_player_connection(player, self) def connect_to(self, player): "Connect the user to this channel" self.typeclass.pre_join_channel(player) if not self.access(player, 'listen'): return False connect = self.typeclass.pre_join_channel(player) if not connect: return False player = player.dbobj conn = PlayerChannelConnection.objects.create_connection(player, self) if conn: self.typeclass.post_join_channel(player) return True return False def disconnect_from(self, player): "Disconnect user from this channel." disconnect = self.typeclass.pre_leave_channel(self, player) if not disconnect: return False PlayerChannelConnection.objects.break_connection(player, self) self.typeclass.post_leave_channel(self, player) return True def delete(self): "Clean out all connections to this channel and delete it." for connection in ChannelDB.objects.get_all_connections(self): connection.delete() super(ChannelDB, self).delete() def access(self, accessing_obj, access_type='listen', default=False): """ Determines if another object has permission to access. accessing_obj - object trying to access this one access_type - type of access sought default - what to return if no lock of access_type was found """ return self.locks.check(accessing_obj, access_type=access_type, default=default) class PlayerChannelConnection(SharedMemoryModel): """ This connects a player object to a particular comm channel. The advantage of making it like this is that one can easily break the connection just by deleting this object. """ # Player connected to a channel db_player = models.ForeignKey("players.PlayerDB", verbose_name='player') # Channel the player is connected to db_channel = models.ForeignKey(ChannelDB, verbose_name='channel') # Database manager objects = managers.PlayerChannelConnectionManager() # player property (wraps db_player) #@property def player_get(self): "Getter. Allows for value = self.player" return self.db_player #@player.setter def player_set(self, value): "Setter. Allows for self.player = value" self.db_player = value self.save() #@player.deleter def player_del(self): "Deleter. Allows for del self.player. Deletes connection." self.delete() player = property(player_get, player_set, player_del) # channel property (wraps db_channel) #@property def channel_get(self): "Getter. Allows for value = self.channel" return self.db_channel.typeclass #@channel.setter def channel_set(self, value): "Setter. Allows for self.channel = value" self.db_channel = value.dbobj self.save() #@channel.deleter def channel_del(self): "Deleter. Allows for del self.channel. Deletes connection." self.delete() channel = property(channel_get, channel_set, channel_del) def __str__(self): return "Connection Player '%s' <-> %s" % (self.player, self.channel) class Meta: "Define Django meta options" verbose_name = "Channel<->Player link" verbose_name_plural = "Channel<->Player links" class ExternalChannelConnection(SharedMemoryModel): """ This defines an external protocol connecting to a channel, while storing some critical info about that connection. """ # evennia channel connecting to db_channel = models.ForeignKey(ChannelDB, verbose_name='channel', help_text='which channel this connection is tied to.') # external connection identifier db_external_key = models.CharField('external key', max_length=128, help_text='external connection identifier, used by calling protocol.') # eval-code to use when the channel tries to send a message # to the external protocol. db_external_send_code = models.TextField('executable code', blank=True, help_text='this is a custom snippet of Python code to connect the external protocol to the in-game channel.') # custom config for the connection db_external_config = models.TextField('external config', blank=True, help_text='configuration options on form understood by connection.') # activate the connection db_is_enabled = models.BooleanField('is enabled', default=True, help_text='turn on/off the connection.') objects = managers.ExternalChannelConnectionManager() class Meta: verbose_name = "External Channel Connection" def __str__(self): return "%s <-> external %s" % (self.channel.key, self.db_external_key) # channel property (wraps db_channel) #@property def channel_get(self): "Getter. Allows for value = self.channel" return self.db_channel #@channel.setter def channel_set(self, value): "Setter. Allows for self.channel = value" self.db_channel = value self.save() #@channel.deleter def channel_del(self): "Deleter. Allows for del self.channel. Deletes connection." self.delete() channel = property(channel_get, channel_set, channel_del) # external_key property (wraps db_external_key) #@property def external_key_get(self): "Getter. Allows for value = self.external_key" return self.db_external_key #@external_key.setter def external_key_set(self, value): "Setter. Allows for self.external_key = value" self.db_external_key = value self.save() #@external_key.deleter def external_key_del(self): "Deleter. Allows for del self.external_key. Deletes connection." self.delete() external_key = property(external_key_get, external_key_set, external_key_del) # external_send_code property (wraps db_external_send_code) #@property def external_send_code_get(self): "Getter. Allows for value = self.external_send_code" return self.db_external_send_code #@external_send_code.setter def external_send_code_set(self, value): "Setter. Allows for self.external_send_code = value" self.db_external_send_code = value self.save() #@external_send_code.deleter def external_send_code_del(self): "Deleter. Allows for del self.external_send_code. Deletes connection." self.db_external_send_code = "" self.save() external_send_code = property(external_send_code_get, external_send_code_set, external_send_code_del) # external_config property (wraps db_external_config) #@property def external_config_get(self): "Getter. Allows for value = self.external_config" return self.db_external_config #@external_config.setter def external_config_set(self, value): "Setter. Allows for self.external_config = value" self.db_external_config = value self.save() #@external_config.deleter def external_config_del(self): "Deleter. Allows for del self.external_config. Deletes connection." self.db_external_config = "" self.save() external_config = property(external_config_get, external_config_set, external_config_del) # is_enabled property (wraps db_is_enabled) #@property def is_enabled_get(self): "Getter. Allows for value = self.is_enabled" return self.db_is_enabled #@is_enabled.setter def is_enabled_set(self, value): "Setter. Allows for self.is_enabled = value" self.db_is_enabled = value self.save() #@is_enabled.deleter def is_enabled_del(self): "Deleter. Allows for del self.is_enabled. Deletes connection." self.delete() is_enabled = property(is_enabled_get, is_enabled_set, is_enabled_del) # # methods # def to_channel(self, message, *args, **kwargs): "Send external -> channel" if 'from_obj' in kwargs and kwargs.pop('from_obj'): from_obj = self.external_key self.channel.msg(message, senders=[self], *args, **kwargs) def to_external(self, message, senders=None, from_channel=None): "Send channel -> external" # make sure we are not echoing back our own message to ourselves # (this would result in a nasty infinite loop) print senders if self in make_iter(senders):#.external_key: return try: # we execute the code snippet that should make it possible for the # connection to contact the protocol correctly (as set by the protocol). # Note that the code block has access to the variables here, such # as message, from_obj and from_channel. exec(to_str(self.external_send_code)) except Exception: logger.log_trace("Channel %s could not send to External %s" % (self.channel, self.external_key))