import os, sys
from datetime import datetime
from dateutil import relativedelta

sys.path.append('/var/www/digisnaxx.ado/scrapers')
import dtss
dtss.getReady()

from time import sleep
from pprint import pprint as ppr
import pytz

from events.models import Organization, Scraper, Calendar, Event
import events.digitools as digitools

tz_str = "-0500 UTC"
DATETIME_FORMAT = '%A, %B %d, %Y %I:%M %p %z %Z'

venue, created = Organization.objects.get_or_create(
     name="Asylum Comedy",
     city="Boston",
     website="https://calendar.improvasylum.com/",
     is_venue = True
    )

scraper,item_count_start,virtcal = digitools.getScraper(venue, venue.website, 'bos')
scraper.items = 0
scraper.save()

def get_events(ps, event_type):
    contents = ps.xpath('.//*/div[@class="space-y-4"]')
    for c in contents:
        try:
            event = {}
            date = c.xpath('.//*/h3[@class="text-lg font-extrabold uppercase"]/text()')[0]
            time = c.xpath('.//*/p[@class="flex items-center gap-2"]/text()')[0].split("-")[0]
            venue_str = c.xpath('.//*/p[@class="flex items-center gap-2"]/text()')[1]
            event['scraper'] = scraper
            event['calendars'] = [scraper.calendar]
            event['title'] = c.xpath('.//*/h4[@class="text-lg font-extrabold transition-colors"]/text()')[0]
            event['date'] = [date, time, tz_str]
            event['date'] = " ".join(event['date'])
            event['dateStamp'] = datetime.strptime(event['date'], DATETIME_FORMAT)
            event['link'] = venue.website
            digitools.createBasicEvent(event, "Co", venue)
            scraper.items+=1
            print("\nSuccessn")
        except Exception as e:
            print("\nError: ", e)
            pass

if len(sys.argv) >= 2:
    arg1 = sys.argv[1]
    br = digitools.getBrowser(arg1)
else:
    print("No run_env")
    br.close()
    quit()

ps = digitools.getSource(br, venue.website)
get_events(ps, "Co")
sleep(3)

digitools.updateScraper(scraper, item_count_start)
br.close()