不在python中写入文件

Question

I am using selenium to scrape a website and I have stored all the links in a.txt. Now, I wish to scrape individual links from each of those websites and write to b.txt. problem is, my code does not write to the second and file and I do not know why. I am printing the scrapped values to the console and it works perfectly fine. It just doesn't write to the file b.txt

Any ideas what Could've gone wrong? Below is my code.

from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC

file1 = 'a.txt'
file2 = 'b.txt'
xpath = '//*[@id="jw"]/div[2]/video'
driver = webdriver.Chrome()
videos = []
j = 0
so = open(file2, 'w')
with open(file1, 'r') as fo:
    for url in fo:
        driver.get(url)
        wait = EC.presence_of_element_located((By.XPATH, xpath))
        WebDriverWait(driver, 5).until(wait)
        video = driver.find_element_by_xpath(xpath)
        link = str(video.get_attribute('src'))
        so.write(link + '\n')
        videos.append(link)
        j += 1
        print j
        print link
        print videos
so.close()

Answer 1

Do you mean that your code only writes the first time around (looks like some typo in your question that confuses me a bit), but if thats the case, maybe you need to append

open(file1, 'a')

instead of

open(file1, 'r')