music_posting+cover.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### **przyjazny bot**\n",
    "#### post lyrics from top 50 songs in Poland on Twitter"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Import necessary libraries"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import spotipy\n",
    "import spotipy.oauth2 as oauth2\n",
    "from spotipy.oauth2 import SpotifyOAuth\n",
    "from spotipy.oauth2 import SpotifyClientCredentials\n",
    "import lyricsgenius\n",
    "import tweepy\n",
    "import random\n",
    "import os\n",
    "import pandas as pd\n",
    "from dotenv import load_dotenv\n",
    "import requests"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "##### Spotify API Authorization"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "load_dotenv()\n",
    "\n",
    "auth_manager = SpotifyClientCredentials(client_id=os.getenv(\"SPOTIPY_CLIENT_ID\"), client_secret=os.getenv(\"SPOTIPY_CLIENT_SECRET\"))\n",
    "sp = spotipy.Spotify(auth_manager=auth_manager)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Download track ids from specific playlist"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "def download_tracks_ids(username, playlist_id):\n",
    "    tracks_ids = []\n",
    "    playlist = sp.user_playlist(username, playlist_id)\n",
    "    for item in playlist['tracks']['items']:\n",
    "        track = item['track']\n",
    "        tracks_ids.append(track['id'])\n",
    "    return tracks_ids"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "['3Nqxx5fdl3FQlrk3zarRmA', '09m29we7x9tgxyo6f3OQ87', '384lHduuDsrR4HPUwR3fG4', '4JmaOG7bfj1uNv8jEUg0Ce', '3f9Ugg030LDe0p6ospTORC', '3p6uAP2ajkobkIW8IkEpgr', '3nqQXoyQOWXiESFLlDF1hG', '4uUG5RXrOk84mYEfFvj3cK', '73vIOb4Q7YN6HeJTbscRx5', '2yxPqxHbj1L7vrCpHLtAcG', '08sMDBcjUBXWV99gZI092C', '5GdFPcbGDQqtbVO2jl6Ez5', '4nv5o5Xo4ySBaIVnXr75Xs', '1yiGGRGdHEc1F3jF5jEVmH', '561jH07mF1jHuk7KlaeF0s', '11BKm0j4eYoCPPpCONAVwA', '4N6rgD6EqLbWcl5ld3NqTP', '0zmxM4MXfisJRTQcPa1wbv', '7jtQIBanIiJOMS6RyCx6jZ', '4lPAZiTJbaQ4irMzcq0GOU', '11JXHNIH2xO5INcJQOWCjY', '53OQFPwyflA0jMyUjUSawr', '0JXXNGljqupsJaZsgSbMZV', '2KTfSSKBBfMLXwDqrtoOE5', '4h9wh7iOZ0GGn8QVp4RAOB', '5YvL3ytkusKcM5ozxD9fKN', '5NbxvSCL1xLqqWLqG6LjwE', '5XeFesFbtLpXzIVDNQP22n', '1cAAn5PKy40T1p15fpTPW8', '1Qrg8KqiBpW07V7PNxwwwL', '14154tJOTjeyeOyhnqP4QM', '2QjOHCTQ1Jl3zawyYOpxh6', '1VoUHMVxu1qG2Z9josOhWs', '2yEWVnp65tTXSrRW03IqXs', '1N8TTK1Uoy7UvQNUazfUt5', '3rb0tMq42WfggucPm0HHkA', '4Dvkj6JhhA12EX05fT7y2e', '5fjcRpwg94xQ2hrSqw6x06', '5fxFhJyveLasVUfuHf6lYP', '2qqiGxy47FbviXuPW7crUH', '46KrcgLn015ammHqC8FWsw', '2dHHgzDwk4BJdRwy9uXhTO', '5YdnOm5990Kfq1Jodws98B', '4nng35mDbhJjPyGwXFNk76', '2nGA88iGanJ3rkDaWXQLyv', '1J3rvvPqfplUP3w1y9y8In', '4zN21mbAuaD0WqtmaTZZeP', '5qBLfeAGlgGGsckKhsMWpF', '5WxVXxCMRnvxUKFq40ELwq', '3zkyus0njMCL6phZmNNEeN']\n"
     ]
    }
   ],
   "source": [
    "#tracks_ids = download_tracks_ids(os.getenv(\"USERNAME_2\"), os.getenv(\"PLAYLIST_ID\")) \n",
    "tracks_ids = download_tracks_ids(os.getenv(\"USERNAME\"), os.getenv(\"PLAYLIST_ID\")) \n",
    "print(tracks_ids)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Draw of a track id"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "2yxPqxHbj1L7vrCpHLtAcG\n"
     ]
    }
   ],
   "source": [
    "track_id = random.choice(tracks_ids)\n",
    "print(track_id)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Collect data (title and main artist) needed to search lyrics via the Genius API"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "def artist_genius(track_id):\n",
    "    track_info = sp.track(track_id)\n",
    "    artist_genius = track_info['album']['artists'][0]['name']\n",
    "    return artist_genius"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "def title(track_id):\n",
    "    song_info = sp.track(track_id)\n",
    "    title = song_info['name']\n",
    "    return title"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Collect information about all artists to include in a tweet"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "def artists(track_id):\n",
    "    track_info = sp.track(track_id)\n",
    "    artists_list = []\n",
    "    for i in range(0, len(track_info['album']['artists'])):\n",
    "        artists_list.append(track_info['album']['artists'][i]['name']) \n",
    "    str_artists = \", \"\n",
    "    return(str_artists.join(artists_list))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Dawid Podsiadło mori Dawid Podsiadło\n"
     ]
    }
   ],
   "source": [
    "print(artist_genius(track_id), title(track_id), artists(track_id))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Get album cover for given song"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "def get_album_cover(track_id):\n",
    "    track_info = sp.track(track_id = track_id)\n",
    "    img_url = track_info['album']['images'][0]['url']\n",
    "    img_data = requests.get(img_url).content\n",
    "    with open('album_cover.jpg', 'wb') as handler:\n",
    "        handler.write(img_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "https://i.scdn.co/image/ab67616d0000b2737b0be3508536803e5ff5e0bb\n"
     ]
    }
   ],
   "source": [
    "get_album_cover(track_id)\n",
    "print(sp.track(track_id = track_id)['album']['images'][0]['url'])"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "##### Genius API"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    "genius = lyricsgenius.Genius(os.getenv(\"GENIUS_CLIENT_ID\"))\n",
    "genius.remove_section_headers = True"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Download track lyrics from Genius"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "def raw_lyrics(track_id):\n",
    "    lyrics = genius.search_song(title(track_id), artist_genius(track_id)).lyrics\n",
    "    return lyrics"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Processing the text to include it in the tweet"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "def tweet_content(lyrics, track_id):\n",
    "    lyrics = lyrics.split('\\n')\n",
    "    for index in range(len(lyrics)):\n",
    "        if lyrics[index] == \"\" or \"[\" in lyrics[index]:\n",
    "            lyrics[index] = \"XXX\"\n",
    "    lyrics = [i for i in lyrics if i != \"XXX\"]\n",
    "\n",
    "    random_num = random.randrange(0, len(lyrics)-1)\n",
    "    tweet = lyrics[random_num] + \"\\n\" + lyrics[random_num+1] + \"\\n\" + \"\\n\" + title(track_id) + \" by \" + artists(track_id)\n",
    "    tweet = tweet.replace(\"\\\\\", \"\")\n",
    "    return tweet"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Searching for \"mori\" by Dawid Podsiadło...\n",
      "Done.\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "'Czekać aż i Ty dołączysz tu\\nNie wolno mi, zwlekać chwili dłużej\\n\\nmori by Dawid Podsiadło'"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "lyrics = raw_lyrics(track_id)\n",
    "tweet_content(lyrics, track_id)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "##### Twitter API"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "First API v1.1 authentication, only to use *media_upload* method (it's not available in API v2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "auth = tweepy.OAuthHandler(os.getenv(\"CONSUMER_KEY\"), os.getenv(\"CONSUMER_SECRET\"))\n",
    "auth.set_access_token(os.getenv(\"ACCESS_TOKEN\"), os.getenv(\"ACCESS_TOKEN_SECRET\"))\n",
    "\n",
    "apiv1 = tweepy.API(auth)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Upload album cover to twitter (we have to do that in order to attach it to tweet later)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "media = apiv1.media_upload(filename = \"album_cover.jpg\")\n",
    "media_id = media.media_id_string #media_id potem trzeba wstawić do metody create_tweet"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "And now API v2 authentication"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "api = tweepy.Client(bearer_token=os.getenv(\"BEARER_TOKEN\"), consumer_key= os.getenv(\"CONSUMER_KEY\"), consumer_secret=os.getenv(\"CONSUMER_SECRET\"), access_token=os.getenv(\"ACCESS_TOKEN\"), access_token_secret=os.getenv(\"ACCESS_TOKEN_SECRET\"))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Tweet lyrics of one of TOP 50 tracks in Poland"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Searching for \"mori\" by Dawid Podsiadło...\n",
      "Done.\n"
     ]
    }
   ],
   "source": [
    "result = None\n",
    "while result is None:\n",
    "    try:\n",
    "      lyrics = raw_lyrics(track_id)\n",
    "      result = api.create_tweet(text=tweet_content(lyrics, track_id), media_ids = [media_id])\n",
    "    except:\n",
    "         pass"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Retweet quote tweets and reply on mention"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Draw of a track id"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "11JXHNIH2xO5INcJQOWCjY\n"
     ]
    }
   ],
   "source": [
    "track_id = random.choice(tracks_ids)\n",
    "print(track_id)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [],
   "source": [
    "def get_tweet(id):\n",
    "    tweet = api.get_tweet(id, expansions=['author_id'], user_fields=['username'])\n",
    "    return tweet"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [],
   "source": [
    "def quote_tweets(query, max_results):\n",
    "    tweets = api.search_recent_tweets(query=query, max_results=max_results)\n",
    "    results = []\n",
    "    if not tweets.data is None and len(tweets.data) > 0:\n",
    "        for tweet in tweets.data:\n",
    "            twt = get_tweet(tweet['id'])\n",
    "            obj = {}\n",
    "            obj['id'] = tweet.id\n",
    "            obj['username'] = twt.includes['users'][0].username\n",
    "            obj['text'] = tweet.text\n",
    "            results.append(obj)\n",
    "    return results"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Check if whether a tweet (id) has already received a reply"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [],
   "source": [
    "def check(id):\n",
    "    with open('last_tweet_ids.txt') as f:\n",
    "        datafile = f.readlines()\n",
    "    for line in datafile:\n",
    "        if id in line:\n",
    "            return True\n",
    "    return False"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Add tweet (id) to database = text file\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [],
   "source": [
    "def append_new_id(text_to_append):\n",
    "    with open('last_tweet_ids.txt', \"a+\") as file_object:\n",
    "        file_object.seek(0)\n",
    "        data = file_object.read(100)\n",
    "        if len(data) > 0:\n",
    "            file_object.write(\"\\n\")\n",
    "        file_object.write(text_to_append)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'id': 1615763990038052872,\n",
       "  'username': 'Szafa9',\n",
       "  'text': '@przyjaznybot draw lyrics xyz'},\n",
       " {'id': 1615763117798981652,\n",
       "  'username': 'przyjaznybot',\n",
       "  'text': 'Jeszcze kilka słów\\nA jeśli już, to wezmę tam ze sobą bukiet róż\\n\\nmori by Dawid Podsiadło https://t.co/s4jzHSH7s8'},\n",
       " {'id': 1615740148905398277,\n",
       "  'username': 'przyjaznybot',\n",
       "  'text': 'hello @Szafa9\\n\\nYou might also like\\nTeraz jest nasz czas! - wybiła ta godzina\\n\\nRarara by Miszel, Kabe https://t.co/3MRXzsLFKp'},\n",
       " {'id': 1615739685589901312,\n",
       "  'username': 'Szafa9',\n",
       "  'text': '@przyjaznybot draw lyrics'},\n",
       " {'id': 1615734073850314753,\n",
       "  'username': 'przyjaznybot',\n",
       "  'text': \"Bit od CatchUp'a, płynąłeś na fali, a jesteś za burtą\\nTyle jest dymu w tym jebanym studio, że mamy tu sufit, a piję pod chmurką\\n\\nclub2020 by club2020, Taco Hemingway, Oki https://t.co/Uf7ILJupjC\"},\n",
       " {'id': 1615489554726748160,\n",
       "  'username': 'przyjaznybot',\n",
       "  'text': 'RT @Marhalus2: @przyjaznybot draw lyrics'},\n",
       " {'id': 1615489414863413248,\n",
       "  'username': 'Marhalus2',\n",
       "  'text': '@przyjaznybot draw lyrics'},\n",
       " {'id': 1615488501847396352,\n",
       "  'username': 'przyjaznybot',\n",
       "  'text': 'RT @Marhalus2: @przyjaznybot  draw lyrics'},\n",
       " {'id': 1615488337405517826,\n",
       "  'username': 'Marhalus2',\n",
       "  'text': '@przyjaznybot  draw lyrics'},\n",
       " {'id': 1615440242290352134,\n",
       "  'username': 'przyjaznybot',\n",
       "  'text': 'Hello :) https://t.co/O3MZDeviVb'}]"
      ]
     },
     "execution_count": 25,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "QUERY = \"przyjaznybot\"\n",
    "MAX_RESULTS = 10\n",
    "\n",
    "quote_tweets(QUERY, MAX_RESULTS)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "From list of dictionaries to list of specific variables"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [],
   "source": [
    "values_of_usernames = [dict['username'] for dict in quote_tweets(QUERY, MAX_RESULTS)]\n",
    "values_of_ids = [dict['id'] for dict in quote_tweets(QUERY, MAX_RESULTS)]\n",
    "values_of_text = [dict['text'] for dict in quote_tweets(QUERY, MAX_RESULTS)]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "If someone share a tweet containing the phrase \"przyjazny draw lyrics\", bot create new tweet with the user's mention. <br>\n",
    "If someone quotes a tweet, bot retweet it."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Searching for \"NA KRAŃCU ŚWIATA\" by 2115...\n",
      "Done.\n"
     ]
    }
   ],
   "source": [
    "lines = []\n",
    "lyrics = raw_lyrics(track_id)\n",
    "\n",
    "get_album_cover(track_id)\n",
    "media = apiv1.media_upload(filename = \"album_cover.jpg\")\n",
    "media_id = media.media_id_string\n",
    "\n",
    "for i in range(len(values_of_ids)):\n",
    "    if check(str(values_of_ids[i])) == False:\n",
    "        if values_of_usernames[i] != \"przyjaznybot\":\n",
    "            if \"draw lyrics\" in values_of_text[i] and check(str(values_of_ids[i])) == False:\n",
    "                reply = \"hello @\" + values_of_usernames[i] + \"\\n\" + \"\\n\" + tweet_content(lyrics, track_id)\n",
    "                api.like(values_of_ids[i])\n",
    "                api.create_tweet(text=reply, media_ids = [media_id])\n",
    "                append_new_id(str(values_of_ids[i]))\n",
    "            else:\n",
    "                api.like(values_of_ids[i])\n",
    "                api.retweet(values_of_ids[i])\n",
    "                append_new_id(str(values_of_ids[i]))"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.8"
  },
  "vscode": {
   "interpreter": {
    "hash": "b90b5480b90dfd82255d68efb607ef96370ef33575f247c89a0b81cbaa1e7b55"
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}