Browse Source

addr.py,obj.py: minor changes

MMGen 7 years ago
parent
commit
dd1300d946
3 changed files with 10 additions and 9 deletions
  1. 1 1
      doc/release-notes/release-notes-v0.9.8.md
  2. 4 4
      mmgen/addr.py
  3. 5 4
      mmgen/obj.py

+ 1 - 1
doc/release-notes/release-notes-v0.9.8.md

@@ -43,4 +43,4 @@ the case, you must change your wallet password to an ASCII one (or export your
 brainwallet to another MMGen wallet format) using an older version of MMGen
 brainwallet to another MMGen wallet format) using an older version of MMGen
 before upgrading.
 before upgrading.
 
 
-[1]: ../../doc/README.mswin.md
+[1]: /doc/README.mswin.md

+ 4 - 4
mmgen/addr.py

@@ -436,12 +436,12 @@ Removed {{}} duplicate WIF key{{}} from keylist (also in {pnm} key-address file
 			e.sec = PrivKey(sha256(sha256(seed).digest()).digest(),compressed=compressed,pubkey_type=pubkey_type)
 			e.sec = PrivKey(sha256(sha256(seed).digest()).digest(),compressed=compressed,pubkey_type=pubkey_type)
 
 
 			if self.gen_addrs:
 			if self.gen_addrs:
-				ph = kg.to_pubhex(e.sec)
-				e.addr = ag.to_addr(ph)
+				pubhex = kg.to_pubhex(e.sec)
+				e.addr = ag.to_addr(pubhex)
 				if gen_viewkey:
 				if gen_viewkey:
-					e.viewkey = ag.to_viewkey(ph)
+					e.viewkey = ag.to_viewkey(pubhex)
 				if gen_wallet_passwd:
 				if gen_wallet_passwd:
-					e.wallet_passwd = ag.to_wallet_passwd(ph)
+					e.wallet_passwd = ag.to_wallet_passwd(e.sec)
 
 
 			if type(self) == PasswordList:
 			if type(self) == PasswordList:
 				e.passwd = unicode(self.make_passwd(e.sec)) # TODO - own type
 				e.passwd = unicode(self.make_passwd(e.sec)) # TODO - own type

+ 5 - 4
mmgen/obj.py

@@ -745,12 +745,15 @@ class MMGenAddrType(str,Hilite,InitErrors,MMGenObject):
 				'compressed':False,
 				'compressed':False,
 				'gen_method':'zcash_z',
 				'gen_method':'zcash_z',
 				'addr_fmt':'zcash_z',
 				'addr_fmt':'zcash_z',
+				'extra_attrs': ('viewkey',),
 				'desc':'Zcash z-address' },
 				'desc':'Zcash z-address' },
 		'M': {  'name':'monero',
 		'M': {  'name':'monero',
 				'pubkey_type':'monero',
 				'pubkey_type':'monero',
 				'compressed':False,
 				'compressed':False,
 				'gen_method':'monero',
 				'gen_method':'monero',
 				'addr_fmt':'monero',
 				'addr_fmt':'monero',
+				'wif_label':'spendkey:',
+				'extra_attrs': ('viewkey','wallet_passwd'),
 				'desc':'Monero address'}
 				'desc':'Monero address'}
 	}
 	}
 	def __new__(cls,s,on_fail='die',errmsg=None):
 	def __new__(cls,s,on_fail='die',errmsg=None):
@@ -766,10 +769,8 @@ class MMGenAddrType(str,Hilite,InitErrors,MMGenObject):
 						setattr(me,k,v[k])
 						setattr(me,k,v[k])
 					assert me in g.proto.mmtypes + ('P',), (
 					assert me in g.proto.mmtypes + ('P',), (
 						"'{}': invalid address type for {}".format(me.name,g.proto.__name__))
 						"'{}': invalid address type for {}".format(me.name,g.proto.__name__))
-					me.extra_attrs = []
-					if me.name in ('monero','zcash_z'): me.extra_attrs += ['viewkey']
-					if me.name == 'monero': me.extra_attrs += ['wallet_passwd']
-					me.wif_label = ('wif:','spendkey:')[me.name=='monero']
+					me.extra_attrs = v['extra_attrs'] if 'extra_attrs' in v else ()
+					me.wif_label   = v['wif_label'] if 'wif_label' in v else 'wif:'
 					return me
 					return me
 			raise ValueError,'not found'
 			raise ValueError,'not found'
 		except Exception as e:
 		except Exception as e: