Ooops.... thought the OP wanted restore GC compatibility.
Yes, a pre-2010 replacement drive (DMS/D2A/D2B/D2C/D2E) would work with Wiikey 2. However, these days it makes much more sense to softmod and go with USB loading. Now I basically use discs just for GC games with audio streaming (with YAOSM).
However, the ribbon cable which goes to the laser pickup is different on D4 drives (small PCB). AFAIK, the new GC-less Wiis use a variation of the D4 which has the Gamecube disc reading components removed. An older drive should work.
New (no-GC) D4 drive on the left (not sure why the image won't display, but the link works):