コード例 #1
0
 def __init__(self):
     super().__init__('tweet_epic')
     self.parser.add_argument('--now',
                              action='store_true',
                              help="Tweet the next image now")
     self.epic = EPIC()
     self.geocoder = GeoNamesGeocoder()
     self.state = {
         'image_queue': {},
         'last_posted_image': datetime(2015, 9, 1),
         'last_post_time': datetime(2015, 9, 1)
     }
     self.post_interval = timedelta(minutes=60)
     self.post_interval_fast = timedelta(minutes=45)
コード例 #2
0
 def __init__(self):
     self.log = logging.getLogger(__name__)
     self.config = ConfigParser.ConfigParser()
     self.config.read('epictweet.conf')
     auth = tweepy.OAuthHandler(self.config.get('twitter', 'api_key'),
                                self.config.get('twitter', 'api_secret'))
     auth.set_access_token(self.config.get('twitter', 'access_key'),
                           self.config.get('twitter', 'access_secret'))
     self.twitter = tweepy.API(auth)
     self.epic = EPIC()
     self.geocoder = GeoNamesGeocoder()
     self.state = {
         'image_queue': {},
         'last_posted_image': datetime(2015, 9, 1),
         'last_post_time': datetime(2015, 9, 1)
     }
     self.post_interval = timedelta(minutes=80)
コード例 #3
0
ファイル: tweet_updates.py プロジェクト: russss/dscovr-epic
 def __init__(self):
     self.log = logging.getLogger(__name__)
     self.config = ConfigParser.ConfigParser()
     self.config.read('epictweet.conf')
     auth = tweepy.OAuthHandler(self.config.get('twitter', 'api_key'),
                                self.config.get('twitter', 'api_secret'))
     auth.set_access_token(self.config.get('twitter', 'access_key'),
                           self.config.get('twitter', 'access_secret'))
     self.twitter = tweepy.API(auth)
     self.epic = EPIC()
     self.geocoder = GeoNamesGeocoder()
     self.state = {'image_queue': {},
                   'last_posted_image': datetime(2015, 9, 1),
                   'last_post_time': datetime(2015, 9, 1)}
     self.post_interval = timedelta(minutes=60)
コード例 #4
0
ファイル: tweet_updates.py プロジェクト: russss/dscovr-epic
class TweetEPIC(object):
    def __init__(self):
        self.log = logging.getLogger(__name__)
        self.config = ConfigParser.ConfigParser()
        self.config.read('epictweet.conf')
        auth = tweepy.OAuthHandler(self.config.get('twitter', 'api_key'),
                                   self.config.get('twitter', 'api_secret'))
        auth.set_access_token(self.config.get('twitter', 'access_key'),
                              self.config.get('twitter', 'access_secret'))
        self.twitter = tweepy.API(auth)
        self.epic = EPIC()
        self.geocoder = GeoNamesGeocoder()
        self.state = {'image_queue': {},
                      'last_posted_image': datetime(2015, 9, 1),
                      'last_post_time': datetime(2015, 9, 1)}
        self.post_interval = timedelta(minutes=60)

    def poll(self):
        try:
            most_recent = sorted(self.state['image_queue'].keys())[-1]
        except IndexError:
            most_recent = self.state['last_posted_image']

        try:
            images = self.epic.get_recent_images(most_recent, 20)
        except (ConnectionError, ValueError):
            self.log.exception("Unable to fetch images")
            images = []

        added = 0
        for image in images:
            if (image['date'] not in self.state['image_queue'] and
               image['date'] > self.state['last_posted_image']):
                self.state['image_queue'][image['date']] = image
                added += 1
        if added > 0:
            self.log.info("Added %s images to queue", added)

        if self.state['last_post_time'] < (datetime.now() - self.post_interval) \
           and len(self.state['image_queue']) > 0:
            try:
                self.do_tweet()
            except ConnectionError:
                self.log.exception("Unable to fetch image file")
            self.save_state()

    def do_tweet(self):
        image_date = sorted(self.state['image_queue'].keys())[0]
        image = self.state['image_queue'][image_date]
        self.log.info("Tweeting an image")

        with tempfile.NamedTemporaryFile(suffix='.png') as imagefile:
            self.fetch_image(image, imagefile)
            self.post_tweet(image, imagefile)

        del self.state['image_queue'][image_date]
        self.state['last_posted_image'] = image_date
        self.state['last_post_time'] = datetime.now()
        self.log.info("One image tweeted, %s left in queue", len(self.state['image_queue']))

    def post_tweet(self, image, imagefile):
        lat = image['coords']['centroid_coordinates']['lat']
        lon = image['coords']['centroid_coordinates']['lon']
        self.log.info("Geocoding %s, %s", lat, lon)

        geocoded = self.geocoder.geocode(lat, lon)
        if geocoded[0] is not None:
            place = " ".join(geocoded)

        datestring = "%s %s%s" % (image['date'].strftime("%H:%M on %A %B"),
                                  image['date'].day, suffix(image['date'].day))

        if geocoded[0] is not None:
            text = "%s, %s" % (datestring, place)
        else:
            text = datestring
        self.twitter.update_with_media(imagefile.name, file=imagefile, status=text,
                                       lat=lat, long=lon)

    def fetch_image(self, image, destfile):
        with tempfile.NamedTemporaryFile(suffix='.png') as downloadfile:
            self.epic.download_image(image['image'], downloadfile)
            process_image(downloadfile.name, destfile.name)

    def run(self):
        logging.basicConfig(level=logging.INFO)

        try:
            with open("./state.pickle", "r") as f:
                self.state = pickle.load(f)
        except IOError:
            self.log.exception("Failure loading state file, resetting")

        self.log.info("Running")
        try:
            while True:
                self.poll()
                sleep(120)
        finally:
            self.save_state()
            self.log.info("Shut down.")

    def save_state(self):
        self.log.info("Saving state...")
        with open("./state.pickle", "w") as f:
            pickle.dump(self.state, f, pickle.HIGHEST_PROTOCOL)
コード例 #5
0
class TweetEPIC(Bot):
    def __init__(self):
        super().__init__('tweet_epic')
        self.parser.add_argument('--now',
                                 action='store_true',
                                 help="Tweet the next image now")
        self.epic = EPIC()
        self.geocoder = GeoNamesGeocoder()
        self.state = {
            'image_queue': {},
            'last_posted_image': datetime(2015, 9, 1),
            'last_post_time': datetime(2015, 9, 1)
        }
        self.post_interval = timedelta(minutes=60)
        self.post_interval_fast = timedelta(minutes=45)

    def poll(self):
        try:
            most_recent = sorted(self.state['image_queue'].keys())[-1]
        except IndexError:
            most_recent = self.state['last_posted_image']

        try:
            images = self.epic.get_recent_images(most_recent, 20)
        except (ConnectionError, ValueError):
            self.log.exception("Unable to fetch images")
            images = []

        added = 0
        for image in images:
            if (image['date'] not in self.state['image_queue']
                    and image['date'] > self.state['last_posted_image']):
                self.state['image_queue'][image['date']] = image
                added += 1
        if added > 0:
            self.log.info("Added %s images to queue", added)

        if len(self.state['image_queue']) > 20:
            to_drop = len(self.state['image_queue']) - 20
            self.log.info("Dropping %s images from queue", to_drop)
            for key in sorted(self.state['image_queue'])[0:to_drop]:
                del self.state['image_queue'][key]

        if len(self.state['image_queue']) > 12:
            interval = self.post_interval_fast
        else:
            interval = self.post_interval

        if self.state['last_post_time'] < (datetime.now() - interval) \
           and len(self.state['image_queue']) > 0:
            try:
                self.do_tweet()
            except ConnectionError:
                self.log.exception("Unable to fetch image file")
            self.save_state()

    def do_tweet(self):
        image_date = sorted(self.state['image_queue'].keys())[0]
        image = self.state['image_queue'][image_date]
        self.log.info("Tweeting an image")

        with tempfile.NamedTemporaryFile(suffix='.png') as imagefile:
            self.fetch_image(image, imagefile)
            self.post_tweet(image, imagefile)

        del self.state['image_queue'][image_date]
        self.state['last_posted_image'] = image_date
        self.state['last_post_time'] = datetime.now()
        self.log.info("One image tweeted, %s left in queue",
                      len(self.state['image_queue']))

    def post_tweet(self, image, imagefile):
        lat = image['centroid_coordinates']['lat']
        lon = image['centroid_coordinates']['lon']
        self.log.info("Geocoding %s, %s", lat, lon)

        geocoded = self.geocoder.geocode(lat, lon)
        if geocoded[0] is not None:
            place = " ".join(geocoded)

        datestring = "%s %s%s" % (image['date'].strftime("%H:%M on %A %B"),
                                  image['date'].day, suffix(image['date'].day))

        if geocoded[0] is not None:
            text = "%s, %s" % (datestring, place)
        else:
            text = datestring
        self.post(text, imagefile=imagefile, lat=lat, lon=lon)

    def fetch_image(self, image, destfile):
        with tempfile.NamedTemporaryFile(suffix='.png') as downloadfile:
            self.epic.download_image(image, downloadfile)
            process_image(downloadfile.name, destfile.name)

    def main(self):
        if self.args.now:
            self.state['last_post_time'] = datetime(2015, 9, 1)
        while True:
            self.poll()
            sleep(120)
コード例 #6
0
class TweetEPIC(object):
    def __init__(self):
        self.log = logging.getLogger(__name__)
        self.config = ConfigParser.ConfigParser()
        self.config.read('epictweet.conf')
        auth = tweepy.OAuthHandler(self.config.get('twitter', 'api_key'),
                                   self.config.get('twitter', 'api_secret'))
        auth.set_access_token(self.config.get('twitter', 'access_key'),
                              self.config.get('twitter', 'access_secret'))
        self.twitter = tweepy.API(auth)
        self.epic = EPIC()
        self.geocoder = GeoNamesGeocoder()
        self.state = {
            'image_queue': {},
            'last_posted_image': datetime(2015, 9, 1),
            'last_post_time': datetime(2015, 9, 1)
        }
        self.post_interval = timedelta(minutes=80)

    def poll(self):
        try:
            most_recent = sorted(self.state['image_queue'].keys())[-1]
        except IndexError:
            most_recent = self.state['last_posted_image']

        try:
            images = self.epic.get_recent_images(most_recent, 20)
        except (ConnectionError, ValueError):
            self.log.exception("Unable to fetch images")
            images = []

        added = 0
        for image in images:
            if (image['date'] not in self.state['image_queue']
                    and image['date'] > self.state['last_posted_image']):
                self.state['image_queue'][image['date']] = image
                added += 1
        if added > 0:
            self.log.info("Added %s images to queue", added)

        if self.state['last_post_time'] < (datetime.now() - self.post_interval) \
           and len(self.state['image_queue']) > 0:
            try:
                self.do_tweet()
            except ConnectionError:
                self.log.exception("Unable to fetch image file")
            self.save_state()

    def do_tweet(self):
        image_date = sorted(self.state['image_queue'].keys())[0]
        image = self.state['image_queue'][image_date]
        self.log.info("Tweeting an image")

        with tempfile.NamedTemporaryFile(suffix='.png') as imagefile:
            self.fetch_image(image, imagefile)
            self.post_tweet(image, imagefile)

        del self.state['image_queue'][image_date]
        self.state['last_posted_image'] = image_date
        self.state['last_post_time'] = datetime.now()
        self.log.info("One image tweeted, %s left in queue",
                      len(self.state['image_queue']))

    def post_tweet(self, image, imagefile):
        lat = image['coords']['centroid_coordinates']['lat']
        lon = image['coords']['centroid_coordinates']['lon']
        self.log.info("Geocoding %s, %s", lat, lon)

        geocoded = self.geocoder.geocode(lat, lon)
        if geocoded[0] is not None:
            place = " ".join(geocoded)

        datestring = "%s %s%s" % (image['date'].strftime("%H:%M on %B"),
                                  image['date'].day, suffix(image['date'].day))

        if geocoded[0] is not None:
            text = "%s, %s" % (datestring, place)
        else:
            text = datestring
        self.twitter.update_with_media(imagefile.name,
                                       file=imagefile,
                                       status=text,
                                       lat=lat,
                                       long=lon)

    def fetch_image(self, image, destfile):
        with tempfile.NamedTemporaryFile(suffix='.png') as downloadfile:
            self.epic.download_image(image['image'], downloadfile)
            process_image(downloadfile.name, destfile.name)

    def run(self):
        logging.basicConfig(level=logging.INFO)

        try:
            with open("./state.pickle", "r") as f:
                self.state = pickle.load(f)
        except IOError:
            self.log.exception("Failure loading state file, resetting")

        self.log.info("Running")
        try:
            while True:
                self.poll()
                sleep(120)
        finally:
            self.save_state()
            self.log.info("Shut down.")

    def save_state(self):
        self.log.info("Saving state...")
        with open("./state.pickle", "w") as f:
            pickle.dump(self.state, f, pickle.HIGHEST_PROTOCOL)