Existe-t-il une fonction intégrée en Python qui remplacerait (ou supprimerait, peu importe) l'extension d'un nom de fichier (s'il en a une)?

Exemple:

print replace_extension('/home/user/somefile.txt', '.jpg')

Dans mon exemple: /home/user/somefile.txtdeviendrait/home/user/somefile.jpg

Je ne sais pas si cela compte, mais j'en ai besoin pour un module SCons que j'écris. (Alors peut-être y a-t-il une fonction spécifique à SCons que je peux utiliser?)

J'aimerais quelque chose de propre . Faire un simple remplacement de chaîne de toutes les occurrences de .txtla chaîne n'est évidemment pas propre. (Cela échouerait si mon nom de fichier est somefile.txt.txt.txt)

Essayez os.path.splitext, il devrait faire ce que vous voulez.

import os
print os.path.splitext('/home/user/somefile.txt')[0]+'.jpg'

En développant la réponse d'AnaPana, comment supprimer une extension à l'aide de pathlib (Python> = 3.4):

>>> from pathlib import Path

>>> filename = Path('/some/path/somefile.txt')

>>> filename_wo_ext = filename.with_suffix('')

>>> filename_replace_ext = filename.with_suffix('.jpg')

>>> print(filename)
/some/path/somefile.ext    

>>> print(filename_wo_ext)
/some/path/somefile

>>> print(filename_replace_ext)
/some/path/somefile.jpg

Comme l'a dit @jethro, splitextc'est la meilleure façon de le faire. Mais dans ce cas, il est assez facile de le diviser vous-même, puisque l'extension doit être la partie du nom de fichier venant après la dernière période:

filename = '/home/user/somefile.txt'
print( filename.rsplit( ".", 1 )[ 0 ] )
# '/home/user/somefile'

Le rsplitdit à Python d'effectuer le fractionnement de chaîne en commençant par la droite de la chaîne, et le 1dit d'effectuer au plus un fractionnement (de sorte que par exemple 'foo.bar.baz'-> [ 'foo.bar', 'baz' ]). Puisque rsplitretournera toujours un tableau non vide, nous pouvons en toute sécurité l'indexer 0pour obtenir le nom de fichier moins l'extension.

Je préfère l'approche à une ligne suivante en utilisant str.rsplit () :

my_filename.rsplit('.', 1)[0] + '.jpg'

Exemple:

>>> my_filename = '/home/user/somefile.txt'
>>> my_filename.rsplit('.', 1)
>>> ['/home/user/somefile', 'txt']

Pour Python> = 3.4:

from pathlib import Path

filename = '/home/user/somefile.txt'

p = Path(filename)
new_filename = p.parent.joinpath(p.stem + '.jpg') # PosixPath('/home/user/somefile.jpg')
new_filename_str = str(new_filename) # '/home/user/somefile.jpg'

Gérer plusieurs extensions

Dans le cas où vous avez plusieurs extensions, ce one-liner utilise pathlibet str.replacefonctionne un régal:

Supprimer / supprimer les extensions

>>> from pathlib import Path
>>> p = Path("/path/to/myfile.tar.gz")
>>> str(p).replace("".join(p.suffixes), "")
'/path/to/myfile'

Remplacer les extensions

>>> p = Path("/path/to/myfile.tar.gz")
>>> new_ext = ".jpg"
>>> str(p).replace("".join(p.suffixes), new_ext)
'/path/to/myfile.jpg'

Si vous voulez également une pathlibsortie d'objet, vous pouvez évidemment envelopper la ligne dansPath()

>>> Path(str(p).replace("".join(p.suffixes), ""))
PosixPath('/path/to/myfile')

Tout emballer dans une fonction

from pathlib import Path
from typing import Union

PathLike = Union[str, Path]


def replace_ext(path: PathLike, new_ext: str = "") -> Path:
    extensions = "".join(Path(path).suffixes)
    return Path(str(p).replace(extensions, new_ext))


p = Path("/path/to/myfile.tar.gz")
new_ext = ".jpg"

assert replace_ext(p, new_ext) == Path('/path/to/myfile.jpg')
assert replace_ext(str(p), new_ext) == Path('/path/to/myfile.jpg')
assert replace_ext(p) == Path('/path/to/myfile')

Une autre façon de faire est d'utiliser la str.rpartition(sep)méthode.

Par exemple:

filename = '/home/user/somefile.txt'
(prefix, sep, suffix) = filename.rpartition('.')

new_filename = prefix + '.jpg'

print new_filename