#!/usr/bin/env python2 # -*-coding:UTF-8 -* """ The ZMQ_Sub_Attribute Module ============================ This module is consuming the Redis-list created by the ZMQ_PubSub_Line_Q Module. It perform a sorting on the line's length and publish/forward them to differents channels: *Channel 1 if max length(line) < max *Channel 2 if max length(line) > max The collected informations about the processed pastes (number of lines and maximum length line) are stored in Redis. ..note:: Module ZMQ_Something_Q and ZMQ_Something are closely bound, always put the same Subscriber name in both of them. Requirements ------------ *Need running Redis instances. (LevelDB & Redis) *Need the ZMQ_PubSub_Line_Q Module running to be able to work properly. """ import redis import time from packages import Paste from pubsublogger import publisher import Helper if __name__ == "__main__": publisher.channel = "Script" config_section = 'PubSub_Global' config_channel = 'channel' subscriber_name = 'attributes' h = Helper.Redis_Queues(config_section, config_channel, subscriber_name) # Subscriber h.zmq_sub(config_section) # REDIS # r_serv = redis.StrictRedis( host=h.config.get("Redis_Data_Merging", "host"), port=h.config.getint("Redis_Data_Merging", "port"), db=h.config.getint("Redis_Data_Merging", "db")) # FUNCTIONS # publisher.info("""ZMQ Attribute is Running""") while True: try: message = h.redis_rpop() if message is not None: PST = Paste.Paste(message.split(" ", -1)[-1]) else: if h.redis_queue_shutdown(): print "Shutdown Flag Up: Terminating" publisher.warning("Shutdown Flag Up: Terminating.") break publisher.debug("Script Attribute is idling 10s") time.sleep(10) continue encoding = PST._get_p_encoding() language = PST._get_p_language() PST.save_attribute_redis(r_serv, "p_encoding", encoding) PST.save_attribute_redis(r_serv, "p_language", language) r_serv.sadd("Pastes_Objects", PST.p_path) PST.save_all_attributes_redis(r_serv) except IOError: print "CRC Checksum Failed on :", PST.p_path publisher.error('{0};{1};{2};{3};{4}'.format( "Duplicate", PST.p_source, PST.p_date, PST.p_name, "CRC Checksum Failed"))