polen_avila.py

import requests, json, tweepy, os, random, xmltodict, datetime
from dotenv import load_dotenv

load_dotenv()

CONSUMER_KEY = os.getenv("CONSUMER_KEY_AVILA")
CONSUMER_SECRET = os.getenv("CONSUMER_SECRET_AVILA")
ACCESS_TOKEN = os.getenv("ACCESS_TOKEN_AVILA")
ACCESS_TOKEN_SECRET = os.getenv("ACCESS_TOKEN_SECRET_AVILA")
BEARER_TOKEN = os.getenv("BEARER_TOKEN_AVILA")

POLEN_URL = os.getenv("POLEN_URL_AVILA")

emojis = [
  "\U0001f927", # sneeze
  "\U0001F33E", # wheat
  "\U0001F33A", # flower
  "\U0001f4e2", # loud speaker
  "\U0001F514", # bell
  "\U0001F509", # speaker
  "\U0001F4D6", # open book
  "\U0001F5DE", # newspaper
  "\U0001F4EC"  # mailbox
]

levels = {
  "bajo": "\U0001F7E2", # green
  "medio": "\U0001F7E1", # yellow
  "alto": "\U0001F7E0", # orange
  "muyalto": "\U0001F534" # red
}

hashtag = "#Ávila"

try:
  page = requests.get(POLEN_URL)
  page.raise_for_status()
except requests.exceptions.HTTPError as err:
  raise SystemExit(err)

doc = xmltodict.parse(page.content)

dataDict = {}

dataDict['ciudad'] = "Ávila"
dataDict['fecha-inicio'] = doc["document"]["list"]["element"]["@fecha"]
dataDict['fecha-fin'] = (datetime.datetime.strptime(dataDict['fecha-inicio'], '%d/%m/%Y') + datetime.timedelta(days=7)).strftime('%d/%m/%Y')
dataDict['datos'] = []

for estacion in doc["document"]["list"]["element"]["estacion"]:
  if estacion["@nombre"] == "\u00c1VILA":
    tmpdict = {}
    tmpdict['tipo'] = estacion["tipo_polinico"]["@nombre"].capitalize()
    tmpdict['real'] = estacion["tipo_polinico"]["valor_real"].lower()
    tmpdict['prevision'] = estacion["tipo_polinico"]["valor_previsto"].lower()
    dataDict['datos'].append(tmpdict)

dataDict['datos'] = sorted(dataDict['datos'], key=lambda k: k['tipo']) 

# Open the previous json data
try:
  with open(dataDict['ciudad']+'.json') as json_file:
    previous = json.load(json_file)
except:
  previous = {}

# If the results are the same, do nothing
if previous == dataDict:
  print("Dupe")
  exit()
# Otherwise, save the file for next execution and continue
else:
  with open(dataDict['ciudad']+'.json','w') as json_file:
    json_file.write(json.dumps(dataDict))

#print(json.dumps(dataDict,indent=4))
tweet = random.choice(emojis) + " " + dataDict['fecha-inicio'] + "-" + dataDict['fecha-fin'] + "\n"

for dic in dataDict['datos']:
    for k in dic:
      if k == "tipo":
        tweet += dic[k] + ": "
      elif k == "real":
        if dic[k].startswith("bajo"):
          tweet += levels["bajo"]
        elif dic[k].startswith("moderado"):
          tweet += levels["medio"]
        elif dic[k].startswith("alto"):
          tweet += levels["muyalto"]
      elif k == "prevision":
        if dic[k].startswith("bajo"):
          tweet += "[" + levels["bajo"] + "]\n"
        elif dic[k].startswith("moderado"):
          tweet += "[" + levels["medio"] + "]\n"
        elif dic[k].startswith("alto"):
          tweet += "[" + levels["muyalto"] + "]\n"

tweet += hashtag
print(tweet)

# Authenticate to Twitter
api = tweepy.Client(bearer_token=BEARER_TOKEN)

api = tweepy.Client(
    consumer_key=CONSUMER_KEY, consumer_secret=CONSUMER_SECRET,
    access_token=ACCESS_TOKEN, access_token_secret=ACCESS_TOKEN_SECRET
)

TWEET_LIMIT = 280
if len(tweet.encode('utf-8')) > TWEET_LIMIT:
  lines = tweet.splitlines()
  to_tweet = ""
  posted = ""
  for i in range(len(lines)):
    if len(to_tweet.encode('utf-8')) > TWEET_LIMIT:
      # Remove the last 5 characters to fit (...)
      to_tweet = to_tweet[:-5]
      # Find the latest '\n', remove the leftovers and add (...)
      to_tweet = to_tweet[:to_tweet.rfind('\n')] + "(...)" 
      if posted:
        posted = api.create_tweet(text=to_tweet,in_reply_to_tweet_id=posted.data.get('id'))
      else:
        posted = api.create_tweet(text=to_tweet)
      to_tweet = (lines[i-1] + '\n' + lines[i] + '\n')
    else:
      to_tweet += (lines[i] + '\n')
  api.create_tweet(text=to_tweet,in_reply_to_tweet_id=posted.data.get('id'))
else:
  api.create_tweet(text=tweet)