automate movie night

pull/199/head
Aevann 2023-09-14 01:39:47 +03:00
parent ee990aeaec
commit 9b75debb2d
4 changed files with 105 additions and 7 deletions

View File

@ -26,6 +26,7 @@ RUN mkdir /dm_media
RUN mkdir /songs RUN mkdir /songs
RUN mkdir /temp_songs RUN mkdir /temp_songs
RUN mkdir /videos RUN mkdir /videos
RUN mkdir /orgies
RUN mkdir /audio RUN mkdir /audio
RUN mkdir /asset_submissions RUN mkdir /asset_submissions
RUN mkdir /asset_submissions/emojis RUN mkdir /asset_submissions/emojis

View File

@ -33,3 +33,4 @@ from .media import *
from .push_subscriptions import * from .push_subscriptions import *
from .group import * from .group import *
from .group_membership import * from .group_membership import *
from .orgy import *

View File

@ -1,5 +1,6 @@
import time import time
import os import os
import glob
from sys import stdout from sys import stdout
from shutil import make_archive from shutil import make_archive
from hashlib import md5 from hashlib import md5
@ -22,12 +23,13 @@ from files.helpers.config.const import *
from files.helpers.get import * from files.helpers.get import *
from files.helpers.lottery import check_if_end_lottery_task from files.helpers.lottery import check_if_end_lottery_task
from files.helpers.roulette import spin_roulette_wheel from files.helpers.roulette import spin_roulette_wheel
from files.helpers.sanitize import filter_emojis_only, sanitize
from files.helpers.useractions import * from files.helpers.useractions import *
from files.cli import app, db_session, g from files.cli import app, db_session, g
CRON_CACHE_TIMEOUT = 172800 CRON_CACHE_TIMEOUT = 172800
def cron_fn(every_5m, every_1d, every_fri): def cron_fn(every_5m, every_1d, every_fri_12, every_fri_23, every_sat_00, every_sat_03):
with app.app_context(): with app.app_context():
g.db = db_session() g.db = db_session()
g.v = None g.v = None
@ -76,8 +78,22 @@ def cron_fn(every_5m, every_1d, every_fri):
cache.set('stats', stats.stats(), timeout=CRON_CACHE_TIMEOUT) cache.set('stats', stats.stats(), timeout=CRON_CACHE_TIMEOUT)
g.db.commit() g.db.commit()
if every_fri: if every_fri_12:
print('test!!!', flush=True) _create_1st_post()
g.db.commit()
if every_fri_23:
_create_2nd_post()
g.db.commit()
if every_sat_00:
_create_orgy()
g.db.commit()
if every_sat_03:
_delete_all()
g.db.commit()
except: except:
print(traceback.format_exc(), flush=True) print(traceback.format_exc(), flush=True)
g.db.rollback() g.db.rollback()
@ -89,9 +105,89 @@ def cron_fn(every_5m, every_1d, every_fri):
@app.cli.command('cron', help='Run scheduled tasks.') @app.cli.command('cron', help='Run scheduled tasks.')
@click.option('--every-5m', is_flag=True, help='Call every 5 minutes.') @click.option('--every-5m', is_flag=True, help='Call every 5 minutes.')
@click.option('--every-1d', is_flag=True, help='Call every 1 day.') @click.option('--every-1d', is_flag=True, help='Call every 1 day.')
@click.option('--every-fri', is_flag=True, help='Call every Friday.') @click.option('--every-fri-12', is_flag=True, help='Call every Friday.')
def cron(every_5m, every_1d, every_fri): @click.option('--every-fri-23', is_flag=True, help='Call every Friday.')
cron_fn(every_5m, every_1d, every_fri) @click.option('--every-sat-00', is_flag=True, help='Call every Saturday.')
@click.option('--every-sat-03', is_flag=True, help='Call every Saturday.')
def cron(every_5m, every_1d, every_fri_12, every_fri_23, every_sat_00, every_sat_03):
cron_fn(every_5m, every_1d, every_fri_12, every_fri_23, every_sat_00, every_sat_03)
def get_file():
return max(glob.glob('/orgies/*'), key=os.path.getctime).split('/orgies/')[1]
def get_name():
return get_file().split('.')[0]
def _create_post(title, body):
title += f': {get_name()}'
body += f'''It will be shown [here](/orgy).\nRerun will be Sunday 4 PM EST.\nThere will be a 5-minute bathroom break at the 50:00 mark.'''
title_html = filter_emojis_only(title)
body_html = sanitize(body)
p = Post(
private=False,
notify=True,
author_id=AUTOJANNY_ID,
over_18=False,
new=False,
app_id=None,
is_bot=False,
url='/orgy',
body=body,
body_html=body_html,
embed=None,
title=title,
title_html=title_html,
sub='countryclub',
ghost=False,
chudded=False,
rainbowed=False,
queened=False,
sharpened=False,
stickied_utc=int(time.time())+3600,
stickied="AutoJanny",
distinguish_level=6,
)
g.db.add(p)
cache.delete_memoized(frontlist)
def _create_1st_post():
title = f'Movie Night'
body = f'''Our Movie Night this week will show {get_name()}.\nThe movie will start at 8 PM EST. [Here is a timezone converter for whoever needs it.](https://dateful.com/time-zone-converter?t=8pm&tz1=EST-EDT-Eastern-Time). You can also check this [countdown timer](https://www.tickcounter.com/countdown/4435809/movie-night) instead.\n'''
_create_post(title, body)
def _create_2nd_post():
title = f'Movie Night in 60 minutes'
body = ''
_create_post(title, body)
def _create_orgy():
orgy = Orgy(
title=get_name(),
type='file',
data=f'https://videos.watchpeopledie.tv/orgies/{get_file()}'
)
g.db.add(orgy)
def _delete_all():
orgy = g.db.query(Orgy).one_or_none()
if orgy:
g.db.delete(orgy)
posts = g.db.query(Post).filter_by(author_id=AUTOJANNY_ID, deleted_utc=0).all()
for p in posts:
p.deleted_utc = int(time.time())
p.is_pinned = False
p.stickied = None
g.db.add(p)
cache.delete_memoized(frontlist)
for sort in COMMENT_SORTS.keys():
cache.delete(f'post_{p.id}_{sort}')
def _grant_one_year_badges(): def _grant_one_year_badges():
one_year_ago = int(time.time()) - 364 * 86400 one_year_ago = int(time.time()) - 364 * 86400

View File

@ -56,5 +56,5 @@ if IS_LOCALHOST:
if "cron" not in argv and "load_chat" not in argv: if "cron" not in argv and "load_chat" not in argv:
from files.helpers.cron import cron_fn from files.helpers.cron import cron_fn
print('Starting cron tasks!', flush=True) print('Starting cron tasks!', flush=True)
gevent.spawn(cron_fn, True, False, False) gevent.spawn(cron_fn, True, False, False, False, False, False)
print('Cron tasks Finished!', flush=True) print('Cron tasks Finished!', flush=True)