aboutsummaryrefslogtreecommitdiff
path: root/src/wikiget/file.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/wikiget/file.py')
-rw-r--r--src/wikiget/file.py12
1 files changed, 9 insertions, 3 deletions
diff --git a/src/wikiget/file.py b/src/wikiget/file.py
index c1b9ae6..b890e63 100644
--- a/src/wikiget/file.py
+++ b/src/wikiget/file.py
@@ -15,9 +15,13 @@
# You should have received a copy of the GNU General Public License
# along with Wikiget. If not, see <https://www.gnu.org/licenses/>.
+from mwclient.image import Image
+
+from wikiget import DEFAULT_SITE
+
class File:
- def __init__(self, name, dest=""):
+ def __init__(self, name: str, dest: str = "", site: str = "") -> None:
"""
Initializes a new file with the specified name and an optional destination name.
@@ -26,8 +30,10 @@ class File:
:param dest: destination of the file, if different from the name; if not
specified, defaults to the name
:type dest: str, optional
+ :param site: name of the site hosting the file; if not specified, defaults to
+ the global default site
"""
- self.object = None
- self.site = None
+ self.image: Image = None
self.name = name
self.dest = dest if dest else name
+ self.site = site if site else DEFAULT_SITE