tu devrais modifier ce script comme suit :
# -*- coding: UTF-8 -*-
import subprocess
from bs4 import BeautifulSoup
import requests
import re
import sys
def main(olinks):
headers = {'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537)'}
session = requests.Session()
response = session.get(olinks, headers=headers)
soup = BeautifulSoup(response.content)
oname = soup.find("title")
if oname.text.find('Saison') >= 0:
regexp = r'(.*?\s+-\s+S)aison\s+(\d+)\s+\xc9.*?(\d+)(.*)'
subst = "{title} {season:02d} Ep {episode}"
else:
regexp = r'(.*?\s+-)(\s+)\xc9.*?(\d+)(.*)'
subst = "{title} Ep {episode}"
oname_cleanedup = re.sub(regexp,
lambda m: subst.format(title=m.group(1), season=int(m.group(2)) if m.group(2).find(" ")==-1 else "", episode=m.group(3)),
oname.text)
return oname_cleanedup
if __name__=="__main__":
print main(sys.argv[1])
Comme ça, si tu lance le script directement, tu peux l'utiliser comme avant.
Et si tu veux l'utiliser dans un autre script :
import mon_module
machin = mon_module.main(argument)
Voilà !
Attention, ton argument obligatoire, tu ne le vérifie pas. Là si l'argument n'est pas renseigné, ton programme ... plante
