added loop functionality and removed debug lines

This commit is contained in:
Wardyn 2023-01-28 01:25:08 -08:00
parent 7ec10f4d6d
commit 0437f76af3
1 changed files with 19 additions and 14 deletions

View File

@ -3,7 +3,7 @@ from generate_config import generate_config
import json
import requests
import tempfile
import time
parent = os.path.dirname(os.path.realpath(__file__))
if not os.path.exists(os.path.join(parent, 'config.json')):
@ -14,16 +14,21 @@ with open(os.path.join(parent, 'config.json'), 'r') as config_file:
session = requests.Session()
session.headers.update({"Authorization" : "Bearer " + config['user_token']})
post = requests.get('https://gelbooru.com/index.php', params={'page':'dapi','s':'post', 'q':'index', 'limit':1, 'tags':'loli sort:random score:>=10', 'json':1}).json()['post'][0]
print(post['id'])
with tempfile.TemporaryFile('w+b', suffix='.' + post['file_url'].split('.')[-1]) as image:
#with open('image.' + post['file_url'].split('.')[-1], 'w+b') as image:
image.write(requests.get(post['file_url']).content)
image.seek(0)
attachment = session.post(config['instance'] + '/api/v2/media', files={'file':image}, data={'description': 'gelbooru ID: ' + str(post['id'])}).json()
print(attachment)
content = 'source: https://gelbooru.com/index.php?page=post&s=view&id=' + str(post['id'])
if not 'source_request' in post['tags'].split(' '):
content = 'source: ' + post['source']
sensitive = True
print(session.post(config['instance'] + '/api/v1/statuses', json={'status':content, 'media_ids':[attachment['id']], 'visibility':'direct', 'sensitive': True}).json())
def make_post():
post = requests.get('https://gelbooru.com/index.php', params={'page':'dapi','s':'post', 'q':'index', 'limit':1, 'tags':'loli sort:random score:>=10', 'json':1}).json()['post'][0]
print(post['id'])
with tempfile.TemporaryFile('w+b', suffix='.' + post['file_url'].split('.')[-1]) as image:
#with open('image.' + post['file_url'].split('.')[-1], 'w+b') as image:
image.write(requests.get(post['file_url']).content)
image.seek(0)
attachment = session.post(config['instance'] + '/api/v2/media', files={'file':image}, data={'description': 'gelbooru ID: ' + str(post['id'])}).json()
content = 'source: https://gelbooru.com/index.php?page=post&s=view&id=' + str(post['id'])
if not 'source_request' in post['tags'].split(' '):
content = 'source: ' + post['source']
sensitive = True
session.post(config['instance'] + '/api/v1/statuses', json={'status':content, 'media_ids':[attachment['id']], 'visibility':'unlisted', 'sensitive': True})
while True:
make_post()
time.sleep(1 * 60 * 60)