1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
# -*- coding: utf-8 -*-
""" MapList -- ListWidget controller classes that display information
Copyright © 2009-2010, Michael "Svedrin" Ziegler <diese-addy@funzt-halt.net>
"""
import os
from urllib import urlopen
from PyQt4 import Qt
from PyQt4 import QtGui
class MapListItem( QtGui.QListWidgetItem ):
""" Base class for an Item in a map list widget. """
def __init__( self, parent, map_name, map_exts ):
QtGui.QListWidgetItem.__init__( self, map_name, parent )
self.map_name = map_name
self.map_exts = map_exts
self.setFlags( Qt.Qt.ItemIsSelectable | Qt.Qt.ItemIsUserCheckable | Qt.Qt.ItemIsEnabled )
self.uncheck()
checked = property(
lambda self: self.checkState() == Qt.Qt.Checked,
doc="True if this item has been selected by the user."
)
def check( self ):
""" Set the check state of this item to Checked. """
self.setCheckState( Qt.Qt.Checked )
def uncheck( self ):
""" Set the check state of this item to Unchecked. """
self.setCheckState( Qt.Qt.Unchecked )
class BaseMapList( object ):
""" Base class for a Map list widget controller. """
def __init__( self, widget ):
object.__init__( self )
self.widget = widget
self.widget.setSortingEnabled( True )
def get_map_item( self, mapname ):
""" Get the map item for the map with the given name. """
items = self.widget.findItems( mapname, Qt.Qt.MatchExactly )
if not items:
return None
if len(items) == 1:
return items[0]
else:
raise ValueError( "Found multiple maps." )
def mark_missing( self, other_list ):
""" Mark entries for maps that are missing in other_list by setting the
background color to light gray.
"""
for i in range( self.widget.count() ):
item = self.widget.item(i)
if not other_list.has_map( item.text() ):
item.setBackgroundColor( Qt.Qt.lightGray )
def has_map( self, map_name ):
""" Check if the given map exists in my list. """
for i in range( self.widget.count() ):
item = self.widget.item(i)
if item and item.map_name == map_name:
return True
return False
def choose_missing( self, other_list ):
""" Set CheckState for entries for maps that are missing in other_list to Checked. """
for i in range( self.widget.count() ):
item = self.widget.item(i)
if not other_list.has_map( item.text() ):
item.check()
class ClientMapList( BaseMapList ):
""" Map list widget controller that operates on the local file system. """
def fetch( self, basedir ):
""" Populate the list by finding map files in the given directory. """
while self.widget.takeItem( 0 ):
pass
fileslist = dict()
for thisfile in os.listdir( basedir ):
if not os.path.isfile( "%s/%s" % ( basedir, thisfile ) ):
continue
base, ext = thisfile.split( '.', 1 )
if base not in fileslist:
fileslist[base] = [ext]
else:
fileslist[base].append( ext )
for mapname in fileslist:
if "bsp" in fileslist[mapname]:
self.widget.addItem( MapListItem( self.widget, mapname, fileslist[mapname] ) )
class ServerMapList( BaseMapList ):
""" Map list widget controller that queries the server. """
def fetch( self, baseurl ):
""" Populate the list by querying the server for existing maps. """
while self.widget.takeItem( 0 ):
pass
# Fetch the baseURL - it returns somthing like this:
# dm_lockdown: bsp
# dm_underpass: bsp bsp.bz2
# dm_resistance: bsp
index = urlopen( baseurl ).fp
while True:
line = index.readline()
if not line:
break
map_name, map_exts_string = line.strip().split( ':', 1 )
map_exts = map_exts_string.strip().split( ' ' )
self.widget.addItem( MapListItem( self.widget, map_name, map_exts ) )
|
