aboutsummaryrefslogtreecommitdiffstats
path: root/provision.py
diff options
context:
space:
mode:
Diffstat (limited to 'provision.py')
-rw-r--r--provision.py20
1 files changed, 9 insertions, 11 deletions
diff --git a/provision.py b/provision.py
index b995267..6c97a3c 100644
--- a/provision.py
+++ b/provision.py
@@ -2,19 +2,21 @@ import pandas
import os
import ast
import twint
-import requests
+import requests
+
def download(url):
- filename = url.split('/')[-1]
+ filename = url.split("/")[-1]
r = requests.get(url, allow_redirects=True)
- with open("pics/"+filename, 'wb') as f:
+ with open("pics/" + filename, "wb") as f:
f.write(r.content)
+
# archive @AceYuriBot for images/sources
c = twint.Config()
c.Username = "AceYuriBot"
-c.Images = True
+c.Images = True
c.Store_csv = True
c.Output = "yuribot.csv"
twint.run.Search(c)
@@ -22,14 +24,10 @@ twint.run.Search(c)
os.makedirs("pics", exist_ok=True)
df = pandas.read_csv("yuribot.csv")
source = (
- df["urls"]
- .apply(lambda x: ast.literal_eval(x))
- .apply(lambda x: x[0] if x else None)
-)
-file_location = df["photos"].apply(
- lambda x: os.path.basename(ast.literal_eval(x)[0])
+ df["urls"].apply(lambda x: ast.literal_eval(x)).apply(lambda x: x[0] if x else None)
)
+file_location = df["photos"].apply(lambda x: os.path.basename(ast.literal_eval(x)[0]))
# save to file where bot will pull data from
pandas.concat([source, file_location], axis=1).to_csv("files.csv")
# download images
-df["photos"].apply(lambda x: download(ast.literal_eval(x)[0])) \ No newline at end of file
+df["photos"].apply(lambda x: download(ast.literal_eval(x)[0]))