get_new_filename.py
c6af74908c2cmaster

get_new_filename.py
View Options

			#!/usr/bin/python
			# Copyright (C) 2018 by the RROMPy authors
			#
			# This file is part of RROMPy.
			#
			# RROMPy is free software: you can redistribute it and/or modify
			# it under the terms of the GNU Lesser General Public License as published by
			# the Free Software Foundation, either version 3 of the License, or
			# (at your option) any later version.
			#
			# RROMPy is distributed in the hope that it will be useful,
			# but WITHOUT ANY WARRANTY; without even the implied warranty of
			# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
			# GNU Lesser General Public License for more details.
			#
			# You should have received a copy of the GNU Lesser General Public License
			# along with RROMPy. If not, see <http://www.gnu.org/licenses/>.
			#

			import os
			import numpy as np

			__all__ = ['getNewFilename']

			def getNewFilename(prefix : str = "", extension : str = "dat") -> str:
			n = len(extension) + 1
			filename = "{}{}.{}".format(prefix, np.random.randint(0, 10), extension)
			while os.path.exists(filename):
			filename = filename[: - n] + "{}.{}".format(np.random.randint(10),
			extension)
			return filename